Update app.py

app.py

@@ -204,13 +204,13 @@ with gr.Blocks() as demo:
                 num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
                 num_tokens_global = num_tokens
                 embedding_learning_rate = gr.Textbox(value="0.00005", label="Embedding optimization: Learning rate", interactive= True )
-                max_emb_train_steps =  gr.Number(value="30", label="embedding optimization: Training steps", interactive= True )
+                max_emb_train_steps =  gr.Number(value="15", label="embedding optimization: Training steps", interactive= True )
                 
                 diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
-                max_diffusion_train_steps = gr.Number(value="30", label="UNet Optimization: Learning rate: Training steps", 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="16", label="Batch size", interactive= True )
-                gradient_accumulation_steps=gr.Number(value="5", label="Gradient accumulation", interactive= True )
+                train_batch_size = gr.Number(value="32", label="Batch size", interactive= True )
+                gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
                 
                 add_button  = gr.Button("Run optimization")
                 def run_optimization_wrapper (