fclong commited on
Commit
c92c0f0
·
1 Parent(s): 242fcec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -1,3 +1,12 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ from transformers import BartTokenizer, BartModel
5
+
6
+ tokenizer = BartTokenizer.from_pretrained('facebook/bart-large')
7
+ model = BartModel.from_pretrained('facebook/bart-large')
8
+
9
+ inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
10
+ outputs = model(**inputs)
11
+
12
+ last_hidden_states = outputs.last_hidden_state