Update app/models.py

app/models.py

@@ -73,7 +73,7 @@ class QAModel:
 
         # Load the tokenizer and model from the local path
         self.tokenizer = AutoTokenizer.from_pretrained(model_path)
-        self.model = AutoModel.from_pretrained(model_path).to(self.device)
+        self.model = AutoModel.from_pretrained(model_path,return_dict=True).to(self.device)
         logger.info(f"Loaded QA model: {self.model_name}")
 
     def inference_qa(self, context: str, question: str):