wzkariampuzha commited on
Commit
4b34829
1 Parent(s): d755670

Update extract_abs.py

Browse files
Files changed (1) hide show
  1. extract_abs.py +5 -2
extract_abs.py CHANGED
@@ -331,8 +331,11 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
331
  st.write(len(results),'abstracts classified as epidemiological.')
332
 
333
  sankey_data = (found, relevant, epidemiologic)
334
-
335
- name_gardID = (search_term_list[-1],GARD_dict[search_term_list[-1]])
 
 
 
336
 
337
  return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name_gardID
338
 
 
331
  st.write(len(results),'abstracts classified as epidemiological.')
332
 
333
  sankey_data = (found, relevant, epidemiologic)
334
+ #Export the name and GARD ID to the ap for better integration on page.
335
+ name = search_term_list[-1]
336
+ #Capitalize the first letter in the disease name
337
+ name[0] = name[0].upper()
338
+ name_gardID = (name, GARD_dict[search_term_list[-1]])
339
 
340
  return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name_gardID
341