Update main.py

main.py

@@ -9,12 +9,7 @@ from transformers import pipeline, DistilBertTokenizer, DistilBertForQuestionAns
 
 app = FastAPI()
 
-# Load DistilBERT-based question answering model
-model_name = "distilbert-base-cased-distilled-squad"
-tokenizer = DistilBertTokenizer.from_pretrained(model_name)
-model = DistilBertForQuestionAnswering.from_pretrained(model_name)
-
-nlp_qa = pipeline("question-answering", model=model, tokenizer=tokenizer)
+nlp_qa = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
 description = """
 ## Image-based Document QA