Update utils/target_classifier.py

utils/target_classifier.py

@@ -107,12 +107,12 @@ def target_classification(haystack_doc:pd.DataFrame,
                             
     haystack_doc['Target Label'] = 'NA'
 
-    if not classifier_model:
+    if not target_classifier_model:
 
-        classifier_model = st.session_state['target_classifier']
+        target_classifier_model = st.session_state['target_classifier']
 
         # Get predictions
-        predictions = classifier_model(list(haystack_doc.text))
+        predictions = target_classifier_model(list(haystack_doc.text))
         st.write('predictions')
         st.write(predictions[:10])