wzkariampuzha commited on
Commit
2ad3bb7
1 Parent(s): 71c29ef

Update extract_abs.py

Browse files
Files changed (1) hide show
  1. extract_abs.py +2 -2
extract_abs.py CHANGED
@@ -308,7 +308,7 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
308
  if len(pmid_abs)==0:
309
  st.error('No results were gathered. Enter a new search term.')
310
  else:
311
- gathered, relevant = sankey_initial
312
  epidemiologic = 0
313
  i = 0
314
  my_bar = st.progress(i)
@@ -328,7 +328,7 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
328
  i+=1
329
  my_bar.progress(min(round(i*percent_at_step/100,1),1.0))
330
 
331
- sankey_data = (gathered, relevant,epidemiologic)
332
  st.write(len(results),'abstracts classified as epidemiological.')
333
  return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data
334
 
 
308
  if len(pmid_abs)==0:
309
  st.error('No results were gathered. Enter a new search term.')
310
  else:
311
+ found, relevant = sankey_initial
312
  epidemiologic = 0
313
  i = 0
314
  my_bar = st.progress(i)
 
328
  i+=1
329
  my_bar.progress(min(round(i*percent_at_step/100,1),1.0))
330
 
331
+ sankey_data = (found, relevant, epidemiologic)
332
  st.write(len(results),'abstracts classified as epidemiological.')
333
  return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data
334