Update app.py

app.py

@@ -209,7 +209,7 @@ with gr.Blocks() as demo:
                 diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
                 max_diffusion_train_steps = gr.Number(value="10", label="UNet Optimization: Learning rate: Training steps", interactive= True )
                 
-                train_batch_size = gr.Number(value="32", label="Batch size", interactive= True )
+                train_batch_size = gr.Number(value="20", label="Batch size", interactive= True )
                 gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
                 
                 add_button  = gr.Button("Run optimization")