sidphbot commited on
Commit
ef8272b
1 Parent(s): 04e4f94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -8,10 +8,15 @@ from src.Surveyor import Surveyor
8
 
9
 
10
 
11
- @st.experimental_singleton(suppress_st_warning=True)
12
  def get_surveyor_instance(_print_fn, _survey_print_fn):
13
  with st.spinner('Loading The-Researcher ...'):
14
- return Surveyor(print_fn=_print_fn, survey_print_fn=_survey_print_fn, high_gpu=True)
 
 
 
 
 
 
15
 
16
 
17
  def run_survey(surveyor, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):
 
8
 
9
 
10
 
 
11
  def get_surveyor_instance(_print_fn, _survey_print_fn):
12
  with st.spinner('Loading The-Researcher ...'):
13
+ with tempfile.TemporaryDirectory() as td:
14
+ survey_root = Path(td)
15
+ dir_args = {f'{dname}_dir': survey_root / dname for dname in ['pdf', 'txt', 'img', 'tab', 'dump']}
16
+ for d in dir_args.values():
17
+ d.mkdir(exist_ok=True, parents=True)
18
+ dir_args = {k: str(v.resolve()) for k, v in dir_args.items()}
19
+ return Surveyor(print_fn=_print_fn, survey_print_fn=_survey_print_fn, high_gpu=True, **dir_args)
20
 
21
 
22
  def run_survey(surveyor, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):