Spaces:
Sleeping
Sleeping
seawolf2357
commited on
Commit
โข
4f123ae
1
Parent(s):
9c5b5ad
Update app.py
Browse files
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 = ['
|
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 = '#
|
40 |
-
header_text = "## Arxiv
|
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:
|