theatticusproject/cuad-qa
Updated • 3.89k • 67
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", device_map="auto") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("marshmellow77/roberta-base-cuad")
model = AutoModelForQuestionAnswering.from_pretrained("marshmellow77/roberta-base-cuad", device_map="auto")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: