Update app.py

app.py

@@ -339,7 +339,7 @@ with gr.Blocks() as demo:
         lime_img = gr.Image(label="LIME Explanation")
 
     agent_btn.click(fn=analyze_data, inputs=[file_input], outputs=[insights_output, visual_output])
-    train_btn.click(fn=train_model, inputs=[], outputs=[metrics_output, trials_output])
+    train_btn.click(fn=train_model, inputs=[file_input], outputs=[metrics_output, trials_output])
     explain_btn.click(fn=explainability, inputs=[], outputs=[shap_img, lime_img])
 
 demo.launch(debug=True)