nielsr HF staff commited on
Commit
7652e0c
1 Parent(s): 51f9983

Fix code example

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -26,8 +26,8 @@ Here is how to use this model in PyTorch:
26
  ```python
27
  from transformers import BartTokenizer, BartModel
28
 
29
- tokenizer = BartTokenizer.from_pretrained('facebook/bart-large')
30
- model = BartModel.from_pretrained('facebook/bart-large')
31
 
32
  inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
33
  outputs = model(**inputs)
 
26
  ```python
27
  from transformers import BartTokenizer, BartModel
28
 
29
+ tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
30
+ model = BartModel.from_pretrained('facebook/bart-base')
31
 
32
  inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
33
  outputs = model(**inputs)