Spaces:
Sleeping
Sleeping
Update services/qa_service/qna.py
Browse files
services/qa_service/qna.py
CHANGED
@@ -53,9 +53,9 @@ class QAService:
|
|
53 |
"""Query pinecone outputs and infer results"""
|
54 |
full_context = self.retrieve_context()
|
55 |
|
56 |
-
|
57 |
|
58 |
-
|
59 |
-
|
60 |
|
61 |
return self.question, full_context, # output, context
|
|
|
53 |
"""Query pinecone outputs and infer results"""
|
54 |
full_context = self.retrieve_context()
|
55 |
|
56 |
+
context = [i["text"] for i in full_context]
|
57 |
|
58 |
+
prompt = format_prompt(self.question, context)
|
59 |
+
output = self.model_pipeline.infer(prompt)
|
60 |
|
61 |
return self.question, full_context, # output, context
|