Update app.py

app.py

@@ -271,8 +271,9 @@ with gr.Blocks() as demo:
 
 
     agent_btn.click(fn=run_agent, inputs=[], outputs=[insights_output, visual_output])
-    train_btn.click(fn=train_model, inputs=df_output, outputs=[metrics_output, trials_output])
-    explain_btn.click(fn=explainability, inputs=df_output, outputs=[shap_img, lime_img])
+    train_btn.click(fn=train_model, inputs=[], outputs=[metrics_output, trials_output])
+    explain_btn.click(fn=explainability, inputs=[], outputs=[shap_img, lime_img])
+
 
 
 demo.launch(debug=True)