Update appStore/category.py
Browse files- appStore/category.py +1 -1
appStore/category.py
CHANGED
@@ -32,6 +32,6 @@ def app():
|
|
32 |
classifier = load_categoryClassifier(classifier_name=params['model_name'])
|
33 |
st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
|
34 |
|
35 |
-
df =
|
36 |
threshold= params['threshold'])
|
37 |
st.session_state.key1 = df
|
|
|
32 |
classifier = load_categoryClassifier(classifier_name=params['model_name'])
|
33 |
st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
|
34 |
|
35 |
+
df = category_classification(haystack_doc=df,
|
36 |
threshold= params['threshold'])
|
37 |
st.session_state.key1 = df
|