Spaces:
Build error
Build error
arxiv id list support
Browse files- app.py +1 -1
- src/Surveyor.py +1 -4
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from streamlit_tags import st_tags_sidebar
|
|
| 9 |
@st.experimental_singleton
|
| 10 |
def get_surveyor_instance(_print_fn, _survey_print_fn):
|
| 11 |
with st.spinner('Loading The-Surveyor ...'):
|
| 12 |
-
return Surveyor(_print_fn, _survey_print_fn,
|
| 13 |
|
| 14 |
|
| 15 |
def run_survey(surveyor, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):
|
|
|
|
| 9 |
@st.experimental_singleton
|
| 10 |
def get_surveyor_instance(_print_fn, _survey_print_fn):
|
| 11 |
with st.spinner('Loading The-Surveyor ...'):
|
| 12 |
+
return Surveyor(_print_fn, _survey_print_fn, high_gpu=True)
|
| 13 |
|
| 14 |
|
| 15 |
def run_survey(surveyor, download_placeholder, research_keywords=None, arxiv_ids=None, max_search=None, num_papers=None):
|
src/Surveyor.py
CHANGED
|
@@ -506,10 +506,7 @@ class Surveyor:
|
|
| 506 |
for p in papers:
|
| 507 |
for section in p['sections']:
|
| 508 |
if len(section['highlights']) > 0:
|
| 509 |
-
|
| 510 |
-
content = self.generate_title(section['highlights'])
|
| 511 |
-
else:
|
| 512 |
-
content = self.extractive_summary(''.join(section['highlights']))
|
| 513 |
docs.append(content)
|
| 514 |
selected_pids = [p['id'] for p in papers]
|
| 515 |
meta_abs = []
|
|
|
|
| 506 |
for p in papers:
|
| 507 |
for section in p['sections']:
|
| 508 |
if len(section['highlights']) > 0:
|
| 509 |
+
content = self.extractive_summary(''.join(section['highlights']))
|
|
|
|
|
|
|
|
|
|
| 510 |
docs.append(content)
|
| 511 |
selected_pids = [p['id'] for p in papers]
|
| 512 |
meta_abs = []
|