Update app.py

app.py

@@ -54,7 +54,7 @@ def load_and_classify_csv(file, text_field, event_model, threshold):
         labels.append(res["event"])
         scores.append(res["score"])
 
-    df["model_label"] = labels
+    df["event_label"] = labels
     df["model_score"] = scores
     
     # model_confidence = round(mean(scores), 5)   
@@ -150,7 +150,7 @@ def add_query(to_add, history):
 
 def qa_summarise(selected_queries, qa_llm_model, text_field, data_df):
     
-    qa_input_df = data_df[data_df["model_label"] != "none"].reset_index()
+    qa_input_df = data_df[data_df["event_label"] != "none"].reset_index()
     texts = qa_input_df[text_field].to_list()
     
     summary = generate_answer(qa_llm_model, texts, selected_queries[0], selected_queries, mode="multi_summarize")