Update app.py

app.py

@@ -112,6 +112,6 @@ if __name__ == "__main__":
     iface = gr.Interface(
         fn=train_step,
         inputs=[gr.File(label="Upload CSV"), gr.Textbox()],
-        outputs=["text", gradio.File()]
+        outputs=["text", gr.File()]
     )
     iface.launch()