QuestBoard / test
DetectiveShadow's picture
Create test
b3ab3ed verified
raw
history blame contribute delete
334 Bytes
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)