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
|
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
|
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]:
|