File size: 334 Bytes
b3ab3ed |
1 2 3 4 5 6 7 8 |
pip install transformers torch
from transformers import pipeline, AutoModelForQuestionAnswering, AutoTokenizer
# Load the pretrained model and tokenizer for QA
model_name = "distilbert-base-cased-distilled-squad"
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
|