fix add_data call

app.py

@@ -85,14 +85,14 @@ if uploaded_file:
 
             # Add a button to confirm feedback and log it
             if st.button(f"Submit Feedback for Image {i}", key=f'submit-{i}'):
-                feedback_table.add_data({
-                    'image': wandb.Image(image),
-                    'filetype': uploaded_file.type,
-                    'predicted_label': id2label[prediction],
-                    'predicted_label_id': prediction,
-                    'correct_label': correct_label,
-                    'correct_label_id': label2id[correct_label]
-                })
+                feedback_table.add_data(
+                    wandb.Image(image),
+                    uploaded_file.type,
+                    id2label[prediction],
+                    prediction,
+                    correct_label,
+                    label2id[correct_label],
+                )
                 st.success(f"Feedback for Image {i} submitted!")
 
     print(feedback_table)