theatticusproject/cuad-qa
Updated • 2.1k • 66
How to use marshmellow77/roberta-base-cuad with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="marshmellow77/roberta-base-cuad") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("marshmellow77/roberta-base-cuad")
model = AutoModelForQuestionAnswering.from_pretrained("marshmellow77/roberta-base-cuad")This model is the fine-tuned version of "RoBERTa Base" using CUAD dataset https://huggingface.co/datasets/cuad
Link for model checkpoint: https://github.com/TheAtticusProject/cuad
For the use of the model with CUAD: https://github.com/marshmellow77/cuad-demo and https://huggingface.co/spaces/marshmellow77/contract-review
Related blog posts: