sidphbot commited on
Commit
fbd25d3
1 Parent(s): 8442e32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -21,12 +21,12 @@ def get_surveyor_instance(_print_fn, _survey_print_fn):
21
  return Surveyor(print_fn=_print_fn, survey_print_fn=_survey_print_fn, high_gpu=True, **dir_args)
22
 
23
 
24
- def run_survey(surveyor, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):
25
- zip_file_name, survey_file_name = surveyor.survey(research_keywords,
26
- arxiv_ids,
27
- max_search=max_search,
28
- num_papers=num_papers
29
- )
30
  show_survey_download(zip_file_name, survey_file_name, download_placeholder)
31
 
32
 
@@ -81,7 +81,7 @@ if __name__ == '__main__':
81
  submit = st.form_submit_button(label="Submit")
82
  st.sidebar.write('#### execution log:')
83
 
84
- run_kwargs = {'surveyor':get_surveyor_instance(_print_fn=st.sidebar.write, _survey_print_fn=st.write),
85
  'download_placeholder':download_placeholder}
86
  if submit:
87
  if session_data['research_keywords'] != '':
 
21
  return Surveyor(print_fn=_print_fn, survey_print_fn=_survey_print_fn, high_gpu=True, **dir_args)
22
 
23
 
24
+ def run_survey(_print_fn, _survey_print_fn, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):
25
+ zip_file_name, survey_file_name = get_surveyor_instance(_print_fn, _survey_print_fn).survey(research_keywords,
26
+ arxiv_ids,
27
+ max_search=max_search,
28
+ num_papers=num_papers
29
+ )
30
  show_survey_download(zip_file_name, survey_file_name, download_placeholder)
31
 
32
 
 
81
  submit = st.form_submit_button(label="Submit")
82
  st.sidebar.write('#### execution log:')
83
 
84
+ run_kwargs = {'_print_fn':st.sidebar.write, '_survey_print_fn':st.write,
85
  'download_placeholder':download_placeholder}
86
  if submit:
87
  if session_data['research_keywords'] != '':