Spaces:

Gowtham122
/

Chatmodel

Paused

Gowtham122 commited on Mar 12

Commit

c8e3a10

verified ·

1 Parent(s): 2712129

Update app/models.py

Files changed (1) hide show

app/models.py CHANGED Viewed

@@ -112,7 +112,7 @@ def load_qa_pipeline(model_name: str = "Gowtham122/albertqa"):
         local_path=str(MODEL_DIR / model_name.replace("/", "-")),
         cloud_uri=model_name,  # Hugging Face model ID
     )
-    qa_model = QAModel()
     return qa_model
 def inference_qa(qa_pipeline, context: str, question: str):

         local_path=str(MODEL_DIR / model_name.replace("/", "-")),
         cloud_uri=model_name,  # Hugging Face model ID
     )
+    qa_model = QAModel(model_name,model_location)
     return qa_model
 def inference_qa(qa_pipeline, context: str, question: str):