tykiww commited on
Commit
6afbc36
·
verified ·
1 Parent(s): 10cb430

Update services/qa_service/qna.py

Browse files
Files changed (1) hide show
  1. services/qa_service/qna.py +3 -3
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
- #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
 
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