wzkariampuzha commited on
Commit
12d1c26
1 Parent(s): 9f77129

Update extract_abs.py

Browse files
Files changed (1) hide show
  1. extract_abs.py +1 -1
extract_abs.py CHANGED
@@ -285,7 +285,7 @@ def search_term_extraction(search_term:Union[int,str], maxResults:int, filtering
285
  def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search
286
  NER_pipeline:Any, entity_classes:Union[Set[str],List[str]], #for biobert extraction
287
  extract_diseases:bool, GARD_dict:Dict[str,str], max_length:int, #for disease extraction
288
- classify_model_vars:Tuple[Any,Any,Any,Any,Any]) -> Any, Tuple[int,int,int], Tuple[str,str]: #for classification
289
 
290
  #Format of Output
291
  ordered_labels = order_labels(entity_classes)
 
285
  def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search
286
  NER_pipeline:Any, entity_classes:Union[Set[str],List[str]], #for biobert extraction
287
  extract_diseases:bool, GARD_dict:Dict[str,str], max_length:int, #for disease extraction
288
+ classify_model_vars:Tuple[Any,Any,Any,Any,Any]) -> Tuple[Any, Tuple[int,int,int], Tuple[str,str]]: #for classification
289
 
290
  #Format of Output
291
  ordered_labels = order_labels(entity_classes)