Update app/models.py

app/models.py

@@ -108,12 +108,12 @@ class QAModel:
 # Global instance of the QA model
 qa_model = QAModel()
 
-def load_qa_pipeline():
+def load_qa_pipeline(model_name: str = "Gowtham122/albertqa"):
     """
     Load the QA model and tokenizer.
     """
     global qa_model
-    qa_model = QAModel()  # This will automatically call `_load_model` during initialization
+    qa_model = QAModel(model_name)
     return qa_model
 
 def inference_qa(qa_pipeline, context: str, question: str):