Update appStore/classifier.py

appStore/classifier.py

@@ -11,7 +11,7 @@ from setfit import SetFitModel
 from utils.groups_classifier import load_groupsClassifier, groups_classification
 #import logging
 #logger = logging.getLogger(__name__)
-#from utils.config import get_classifier_params
+from utils.config import get_classifier_params
 #from utils.preprocessing import paraLengthCheck
 #from io import BytesIO
 #import xlsxwriter
@@ -20,19 +20,19 @@ from utils.groups_classifier import load_groupsClassifier, groups_classification
 vg_model = SetFitModel.from_pretrained("leavoigt/vulnerable_groups")
 
 
-# Declare all the necessary variables
+# Retrieve the necessary paramaters
 classifier_identifier = 'group_classification'
 params  = get_classifier_params(classifier_identifier)
 
-# def app(): 
+def app(): 
 
-#     ### Main app code ###
-#     with st.container():
-
-#         # Classify groups
-#         df = group_classification(haystack_doc=df, threshold= params['threshold'])
+     ### Main app code ###
+     with st.container():
+         
+         # Classify groups
+         df = group_classification(haystack_doc=df, threshold= params['threshold'])
         
-# def groups_display():
+ def groups_display():
 #     if  'key1' in st.session_state:
 #         df = st.session_state.key1