text / model.py
fclong's picture
Create model.py
3e1b852
raw
history blame contribute delete
No virus
312 Bytes
from transformers import BartTokenizer, BartModel
tokenizer = BartTokenizer.from_pretrained('facebook/bart-large') model = BartModel.from_pretrained('facebook/bart-large')
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") outputs = model(**inputs)
last_hidden_states = outputs.last_hidden_state