harisankar99 commited on
Commit
8366b60
1 Parent(s): 7ad28b7
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -154,12 +154,13 @@ SYS_PROMPT="""You are friendly battery assistant named EVolve made by company by
154
  # )
155
  # return scores, retrieved_examples
156
  def format_prompt(prompt,retrieved_documents,k):
157
- """using the retrieved documents we will prompt the model to generate our responses"""
158
- PROMPT = f"Question:{prompt}\nContext:"
159
- # +Tell me the reference and doi from where you have taken the answer if it is available.
160
- for idx in range(k) :
161
- PROMPT+= f"Reference: "+str(retrieved_documents['title'][idx])+"\n doi: "+str(retrieved_documents['doi'][idx])+"\n Authors:"+str(retrieved_documents['author'][idx])+"\n Page Number:"+str(retrieved_documents['pages'][idx])+"\n Content: "+str(retrieved_documents['text'][idx])+"\n"
162
- return PROMPT
 
163
 
164
 
165
  @spaces.GPU(duration=150)
 
154
  # )
155
  # return scores, retrieved_examples
156
  def format_prompt(prompt,retrieved_documents,k):
157
+ # """using the retrieved documents we will prompt the model to generate our responses"""
158
+ # PROMPT = f"Question:{prompt}\nContext:"
159
+ # # +Tell me the reference and doi from where you have taken the answer if it is available.
160
+ # for idx in range(k) :
161
+ # PROMPT+= f"Reference: "+str(retrieved_documents['title'][idx])+"\n doi: "+str(retrieved_documents['doi'][idx])+"\n Authors:"+str(retrieved_documents['author'][idx])+"\n Page Number:"+str(retrieved_documents['pages'][idx])+"\n Content: "+str(retrieved_documents['text'][idx])+"\n"
162
+ # return PROMPT
163
+ return prompt
164
 
165
 
166
  @spaces.GPU(duration=150)