init

app.py

@@ -72,17 +72,17 @@ try:
             gr.Textbox(label="Model Name (e.g., 'google/t5-efficient-tiny-nh8')"),
             gr.Textbox(label="Dataset Name (e.g., 'imdb')"),
             gr.Textbox(label="HF hub to push to after training"),
-            gr.Slider(minimum=1, maximum=10, default=3, label="Number of Epochs"),
-            gr.Slider(minimum=1, maximum=16, default=4, label="Batch Size"),
-            gr.Slider(minimum=1, maximum=100, default=50, label="Learning Rate (e-5)"),
-            gr.Slider(minimum=1, maximum=100, default=1, label="Gradient accumulation (e-1)"), 
+            gr.Slider(minimum=1, maximum=10, value=3, label="Number of Epochs"),
+            gr.Slider(minimum=1, maximum=16, value=4, label="Batch Size"),
+            gr.Slider(minimum=1, maximum=1000, value=50, label="Learning Rate (e-5)"),
+            gr.Slider(minimum=1, maximum=100, value=1, label="Gradient accumulation (e-1)"), 
         ],
         outputs="text",
         title="Fine-Tune Hugging Face Model",
         description="This interface allows you to fine-tune a Hugging Face model on a specified dataset."
     )
+    # Launch the interface
+    iface.launch()    
 except Exception as e:
     print(f"An error occurred: {str(e)}, TB: {traceback.format_exc()}")
 
-# Launch the interface
-iface.launch()