ccm commited on
Commit
7e7d205
·
verified ·
1 Parent(s): ca3199a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -92,7 +92,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
92
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
93
  prompt = prompt.replace("{{QUERY_GOES_HERE}}", query)
94
 
95
- return prompt, "\n\n## References\n\n"+"\n".join([str(i+1)+". "+ref for i, ref in enumerate(list(set(references)))])
96
 
97
  def postprocess(response: str, bypass_from_preprocessing: str) -> str:
98
  """
 
92
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
93
  prompt = prompt.replace("{{QUERY_GOES_HERE}}", query)
94
 
95
+ return prompt, "\n\n### References\n\n"+"\n".join([str(i+1)+". "+ref for i, ref in enumerate(list(set(references)))])
96
 
97
  def postprocess(response: str, bypass_from_preprocessing: str) -> str:
98
  """