Update app.py

app.py

@@ -13,17 +13,17 @@ def main():
     interface = gr.Interface(
         fn=train_model,
         inputs=[
-            gr.inputs.Textbox(label="Dataset URL"),
-            gr.inputs.Dropdown(choices=model_options, label="Select Model"),
-            gr.inputs.Slider(minimum=1, maximum=10, default=3, label="Epochs"),
-            gr.inputs.Slider(minimum=1, maximum=64, default=16, label="Batch Size"),
-            gr.inputs.Slider(minimum=1e-5, maximum=1e-1, step=1e-5, default=1e-4, label="Learning Rate")
+            gr.Textbox(label="Dataset URL"),
+            gr.Dropdown(choices=model_options, label="Select Model"),
+            gr.Slider(minimum=1, maximum=10, value=3, label="Epochs"),
+            gr.Slider(minimum=1, maximum=64, value=16, label="Batch Size"),
+            gr.Slider(minimum=1e-5, maximum=1e-1, step=1e-5, value=1e-4, label="Learning Rate")
         ],
-        outputs="json",
+        outputs=gr.JSON(),
         title="Transformers Fine Tuner",
         description="Fine-tune pre-trained transformer models on custom datasets."
     )
     interface.launch()
 
 if __name__ == "__main__":
-    main()
+    main()