chansung commited on
Commit
e518ed3
β€’
1 Parent(s): 046ea77
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import gradio as gr
2
 
3
- from init import get_secrets, initialize_data, update_dataframe
4
- from gen.openllm import GradioLLaMA2ChatPPManager, GradioMistralChatPPManager
 
 
 
5
  from gen.gemini_chat import GradioGeminiChatPPManager
6
  from constants.js import (
7
  UPDATE_SEARCH_RESULTS, OPEN_CHAT_IF,
@@ -13,6 +16,8 @@ from background import process_arxiv_ids
13
  from apscheduler.schedulers.background import BackgroundScheduler
14
 
15
  gemini_api_key, hf_token, dataset_repo_id, request_arxiv_repo_id, restart_repo_id = get_secrets()
 
 
16
  titles, date_dict, requested_arxiv_ids_df, arxivid2data = initialize_data(dataset_repo_id, request_arxiv_repo_id)
17
 
18
  from ui import (
 
1
  import gradio as gr
2
 
3
+ from init import (
4
+ get_secrets, initialize_data,
5
+ update_dataframe, initialize_repos
6
+ )
7
+ from gen.openllm import GradioMistralChatPPManager
8
  from gen.gemini_chat import GradioGeminiChatPPManager
9
  from constants.js import (
10
  UPDATE_SEARCH_RESULTS, OPEN_CHAT_IF,
 
16
  from apscheduler.schedulers.background import BackgroundScheduler
17
 
18
  gemini_api_key, hf_token, dataset_repo_id, request_arxiv_repo_id, restart_repo_id = get_secrets()
19
+ initialize_repos(dataset_repo_id, request_arxiv_repo_id, hf_token)
20
+
21
  titles, date_dict, requested_arxiv_ids_df, arxivid2data = initialize_data(dataset_repo_id, request_arxiv_repo_id)
22
 
23
  from ui import (