ccm commited on
Commit
6338c11
·
verified ·
1 Parent(s): 37dba40

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -13,7 +13,7 @@ import transformers # to load an LLM
13
 
14
  # Constants
15
  GREETING = (
16
- "Howdy! I'm an AI agent that uses [retrieval-augmented generation](https://en.wikipedia.org/wiki/Retrieval-augmented_generation)"
17
  "to answer questions about published at [ASME IDETC](https://asmedigitalcollection.asme.org/IDETC-CIE). "
18
  "I always try to cite my sources, but sometimes things get a little weird. "
19
  "What can I tell you about today?"
@@ -89,8 +89,8 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
89
  + title.title()
90
  + "]("
91
  + url
92
- + ").\n>"
93
- + text + "\n\n"
94
  )
95
 
96
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
 
13
 
14
  # Constants
15
  GREETING = (
16
+ "Howdy! I'm an AI agent that uses [retrieval-augmented generation](https://en.wikipedia.org/wiki/Retrieval-augmented_generation) "
17
  "to answer questions about published at [ASME IDETC](https://asmedigitalcollection.asme.org/IDETC-CIE). "
18
  "I always try to cite my sources, but sometimes things get a little weird. "
19
  "What can I tell you about today?"
 
89
  + title.title()
90
  + "]("
91
  + url
92
+ + ")\n"
93
+ # + text + "\n\n"
94
  )
95
 
96
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)