sidphbot commited on
Commit
508782c
1 Parent(s): 48cd509

arxiv id list support

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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, refresh_models=True)
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
- if self.high_gpu:
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 = []