Update utils/target_classifier.py

utils/target_classifier.py

@@ -119,7 +119,7 @@ def target_classification(haystack_doc:pd.DataFrame,
         # Save labels
         haystack_doc['Target Label'] = pred_labels
 
-                            
+        return haystack_doc            
     # logging.info("Working on action/target extraction")
     # if not classifier_model:
 #     #     classifier_model = st.session_state['target_classifier']
@@ -137,4 +137,3 @@ def target_classification(haystack_doc:pd.DataFrame,
 #     # df.index += 1
 #     # # df['Label_def'] = df['Target Label'].apply(lambda i: _lab_dict[i])
 
-     return haystack_doc