Update app.py

app.py

@@ -224,24 +224,6 @@ def train_model(_):
             f.write(report)
         return "Report saved to model_report.md"
 
-fig, ax = plt.subplots(figsize=(6, 4))
-ConfusionMatrixDisplay.from_estimator(best_model, X_test, y_test, ax=ax)
-plt.savefig("confusion_matrix.png")
-wandb.log({"confusion_matrix": wandb.Image("confusion_matrix.png")})
-
-# Inside your layout:
-compare_button = gr.Button("Compare Models")
-compare_output = gr.Dataframe()
-
-compare_button.click(fn=compare_models, outputs=compare_output)
-
-report_button = gr.Button("Generate Report")
-report_status = gr.Textbox()
-
-report_button.click(
-    fn=lambda: generate_report(metrics_df, trials_df, common_errors),
-    outputs=report_status
-)
 
 # Log common misclassifications to wandb
 wandb.log({"common_errors": wandb.Table(dataframe=common_errors)})