seawolf2357 commited on
Commit
4f123ae
โ€ข
1 Parent(s): 9c5b5ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -13,7 +13,7 @@ from utils import get_md_text_abstract, search_cleaner, get_arxiv_live_search
13
 
14
  retrieve_results = 20
15
  show_examples = True
16
- llm_models_to_choose = ['mistralai/Mixtral-8x7B-Instruct-v0.1', 'None']
17
 
18
  token=os.getenv("HF_TOKEN")
19
 
@@ -25,7 +25,8 @@ generate_kwargs = dict(
25
  )
26
 
27
  ## RAG Model
28
- RAG = RAGPretrainedModel.from_index("colbert/indexes/arxiv_colbert")
 
29
 
30
  try:
31
  gr.Info("Setting up retriever, please wait...")
@@ -36,8 +37,8 @@ except:
36
  gr.Warning("Retriever not working!")
37
 
38
  ## Header
39
- mark_text = '# ๐Ÿฉบ๐Ÿ” Search Results\n'
40
- header_text = "## Arxiv Paper Summary With QA Retrieval Augmented Generation \n"
41
 
42
  try:
43
  with open("README.md", "r") as f:
 
13
 
14
  retrieve_results = 20
15
  show_examples = True
16
+ llm_models_to_choose = ['ArXivGPT-Mixtral-8x7B-Instruct-v0.1', 'None']
17
 
18
  token=os.getenv("HF_TOKEN")
19
 
 
25
  )
26
 
27
  ## RAG Model
28
+ RAG = RAGPretrainedModel.from_index("colbert/indexes/arxiv_colbert")#
29
+ # HF ํ™ˆ ํŒŒ์ผ ์œ„์น˜ ๊ฒฝ๋กœ๋ฅผ ์˜๋ฏธํ•จ. colbert/indexes/arxiv_colbert
30
 
31
  try:
32
  gr.Info("Setting up retriever, please wait...")
 
37
  gr.Warning("Retriever not working!")
38
 
39
  ## Header
40
+ mark_text = '# ๐Ÿ”Ž Search Results\n'
41
+ header_text = "## Arxiv ๋…ผ๋ฌธ ์š”์•ฝ / ๋ถ„์„ ๋Œ€ํ™”ํ˜• AI ๐Ÿ’ป ArXivGPT \n"
42
 
43
  try:
44
  with open("README.md", "r") as f: