Update app.py

app.py

@@ -201,7 +201,7 @@ with gr.Blocks() as app:
         )
 
         dataset_list = gr.Dropdown(choices=list_datasets(), label="Select Dataset")
-        epochs = gr.Slider(10, 1024, value=100, step=10, label="Epochs")
+        epochs = gr.Slider(1, 128, value=100, step=10, label="Epochs")
         lr = gr.Slider(0.001, 0.1, value=0.01, step=0.001, label="Learning Rate")
 
         train_button = gr.Button("Train Model")