Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -1,11 +1,11 @@
 import json
 
 class QAService:
-    def __init__(self, conf, pinecone, model, question, context):
+    def __init__(self, conf, pinecone, model_pipeline, question, context):
         self.conf = conf
         self.pc = pinecone['connection']
         self.embedder = pinecone['embedder']
-        self.model = model
+        self.model_pipeline = model_pipeline
         self.question = question
         self.context = context
     
@@ -35,7 +35,7 @@ class QAService:
     def run(self):
         """Query pinecone outputs and infer results"""
         output = self.retrieve_context()
-        output = self.model.infer(output)
+        output = self.model_pipeline.infer(output)
         
         return output