File size: 446 Bytes
d2306ff 64bb5ac 9c16c6b 1847228 fd5ceb2 f34a7c8 fd5ceb2 f34a7c8 d2306ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import DistilBertTokenizer, DistilBertModel
import torch
tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased-distilled-squad')
model = DistilBertModel.from_pretrained('distilbert-base-cased-distilled-squad')
question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet."
inputs = tokenizer(question, text, return_tensors="pt")
with torch.no_grad():
outputs = model(**inputs)
print(outputs) |