leavoigt commited on
Commit
704e039
1 Parent(s): 4cbfb29

Rename utils/group_classifier.py to utils/groups_classifier.py

Browse files
utils/{group_classifier.py → groups_classifier.py} RENAMED
@@ -19,7 +19,7 @@ _lab_dict = {
19
  6: 'Women'}
20
 
21
  @st.cache_resource
22
- def load_groupClassifier(config_file:str = None, classifier_name:str = None):
23
  """
24
  loads the document classifier using haystack, where the name/path of model
25
  in HF-hub as string is used to fetch the model object.Either configfile or
@@ -51,7 +51,7 @@ def load_groupClassifier(config_file:str = None, classifier_name:str = None):
51
 
52
 
53
  @st.cache_data
54
- def group_classification(haystack_doc:pd.DataFrame,
55
  threshold:float = 0.5,
56
  classifier_model:pipeline= None
57
  )->Tuple[DataFrame,Series]:
 
19
  6: 'Women'}
20
 
21
  @st.cache_resource
22
+ def load_groupsClassifier(config_file:str = None, classifier_name:str = None):
23
  """
24
  loads the document classifier using haystack, where the name/path of model
25
  in HF-hub as string is used to fetch the model object.Either configfile or
 
51
 
52
 
53
  @st.cache_data
54
+ def groups_classification(haystack_doc:pd.DataFrame,
55
  threshold:float = 0.5,
56
  classifier_model:pipeline= None
57
  )->Tuple[DataFrame,Series]: