Update epi_pipeline.py

epi_pipeline.py

@@ -823,8 +823,7 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
                            epi_ner:NER_Pipeline, #for biobert extraction 
                            GARD_Search:GARD_Search, extract_diseases:bool, #for disease extraction
                            epi_classify:Classify_Pipeline) -> pd.DataFrame: #for classification
-    
-    
+   
     #Format of Output
     ordered_labels = order_labels(epi_ner.labels)
     if extract_diseases:
@@ -874,7 +873,7 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
         sankey_data = (found, relevant, epidemiologic)
         #Export the name and GARD ID to the ap for better integration on page.
         name = search_term_list[-1].capitalize()
-        name_gardID = (name, GARD_dict[search_term_list[-1]])
+        name_gardID = (name, GARD_Search.GARD_dict[search_term_list[-1]])
         
         return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name_gardID