Spaces:
Build error
Build error
Update app.py
Browse files
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(
|
| 25 |
-
zip_file_name, survey_file_name =
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 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 = {'
|
| 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'] != '':
|