Update app.py

app.py

@@ -296,10 +296,10 @@ def build_ui():
 
                     with gr.Column(scale=1):
                         gr.Markdown("### Training Hyperparameters")
-                        num_epochs = gr.Slider(1, 10, value=2, step=1, label="Epochs")
+                        num_epochs = gr.Slider(1, 1000, value=2, step=1, label="Epochs")
                         learning_rate = gr.Number(label="Learning Rate", value=2e-4)
                         batch_size = gr.Slider(
-                            1, 32, value=16, step=1, label="Batch Size per Device"
+                            1, 128, value=16, step=1, label="Batch Size per Device"
                         )
                         grad_accum = gr.Slider(
                             1, 64, value=8, step=1, label="Gradient Accumulation Steps"