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)