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)
|