Update app.py

app.py

@@ -36,8 +36,8 @@ def load_and_analyze_csv(file, text_field, event_model):
     flood_related = gr.CheckboxGroup(choices=df[df["model_label"]=="flood"][text_field].to_list())
     not_related = gr.CheckboxGroup(choices=df[df["model_label"]=="none"][text_field].to_list())
     
-    return flood_related, fire_related, not_related, model_confidence, len(df[text_field].to_list()), df, df.columns.tolist()
-
+    return flood_related, fire_related, not_related, model_confidence, len(df[text_field].to_list()), df
+    
 def analyze_selected_texts(selections):
     selected_texts = selections
     
@@ -117,11 +117,10 @@ with gr.Blocks() as demo:
 
         accuracy_button = gr.Button("Calculate Accuracy")
         num_posts = gr.Number(visible=False)
-        cols = gr.CheckboxGroup(value=datacols, visible=False)
-        data = gr.DataFrame(headers=cols) #visible=False
+        data = gr.DataFrame() #visible=False
         
         predict_button.click(load_and_analyze_csv, inputs=[file_input, text_field, event_model], 
-                         outputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, model_confidence, num_posts, data, datacols])
+                         outputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, model_confidence, num_posts, data])
         accuracy_button.click(calculate_accuracy, inputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, num_posts], 
                              outputs=[incorrect, correct, accuracy])