Update appStore/category.py

appStore/category.py

@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 import streamlit as st
-from utils.subtarget_classifier import load_subtargetClassifier, subtarget_classification
+from utils.category_classifier import load_categoryClassifier, category_classification
 import logging
 logger = logging.getLogger(__name__)
 from utils.config import get_classifier_params
@@ -29,7 +29,7 @@ def app():
             df = st.session_state.key1
 
             # Load the classifier model
-            classifier = load_subtargetClassifier(classifier_name=params['model_name'])
+            classifier = load_categoryClassifier(classifier_name=params['model_name'])
             st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
             
             df = netzero_classification(haystack_doc=df,