MarkS commited on
Commit
08a9754
1 Parent(s): 775cfba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -66,8 +66,8 @@ To sum up,my model performs nearly as well as the SOTA rule-based model evaluate
66
 
67
  ```python
68
  from transformers import BartTokenizer, BartForConditionalGeneration
69
- tokenizer = BartTokenizer.from_pretrained("MarkS/QA2D")
70
- model = BartForConditionalGeneration.from_pretrained("MarkS/QA2D")
71
 
72
  input_text = "question: what day is it today? answer: Tuesday"
73
  input = tokenizer(input_text, return_tensors='pt')
 
66
 
67
  ```python
68
  from transformers import BartTokenizer, BartForConditionalGeneration
69
+ tokenizer = BartTokenizer.from_pretrained("MarkS/bart-base-qa2d")
70
+ model = BartForConditionalGeneration.from_pretrained("MarkS/bart-base-qa2d")
71
 
72
  input_text = "question: what day is it today? answer: Tuesday"
73
  input = tokenizer(input_text, return_tensors='pt')