update

utils/methods.py

@@ -59,14 +59,14 @@ def extractive_method(sentJson, sentFeat, model, threshold=0.5, TGB=False):
 
 
     grouped = sentFeat.groupby('section')
-    pred = []
-    true_proba = []
+    pred = np.array([])
+    true_proba = np.array([])
 
     for group_name, group_data in grouped:
-        print(group_name)
         pred_sec, true_proba_sec = predict(group_data)
-        pred.append(pred_sec)
-        true_proba.append(true_proba_sec)
+        # Append to the NumPy arrays
+        pred = np.append(pred, pred_sec)
+        true_proba = np.append(true_proba, true_proba_sec)
 
     body = convert_sentence_df(sentJson, pred, true_proba, TGB)
     res = body[body['predict'] == True]