Update extract_abs.py

extract_abs.py

@@ -285,7 +285,7 @@ def search_term_extraction(search_term:Union[int,str], maxResults:int, filtering
 def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search
                          NER_pipeline:Any, entity_classes:Union[Set[str],List[str]], #for biobert extraction 
                          extract_diseases:bool, GARD_dict:Dict[str,str], max_length:int, #for disease extraction
-                         classify_model_vars:Tuple[Any,Any,Any,Any,Any]) -> pd.DataFrame, Tuple[int,int,int], Tuple[str,str]: #for classification
+                         classify_model_vars:Tuple[Any,Any,Any,Any,Any]) -> Any, Tuple[int,int,int], Tuple[str,str]: #for classification
                                                                                                                                              
     #Format of Output
     ordered_labels = order_labels(entity_classes)