Update app.py

app.py

@@ -242,19 +242,12 @@ with gr.Blocks() as demo:
                             gradient_accumulation_steps=int(gradient_accumulation_steps)
                         )
                         run_optimization()
-                        print('finish')
+                        gr.Info("Optimization Finished! Move to the next step.")
                         return "Optimization finished! Move to the next step.",gr.Button("Run Editing (Check log for progress.)",interactive = True)
                     except:
+                        gr.Warning("CUDA out of memory, use a smaller batch size or try another picture.")
                         return "CUDA out of memory, use a smaller batch size or try another picture.",gr.Button("Run Editing (Check log for progress.)",interactive = False)
 
-                
-                def immediate_update():
-                    return gr.Button("Run Optimization (Check log for progress.).", interactive=True)
-
-                #add_button.click(fn=immediate_update, inputs=[], outputs=[add_button])
-                #add_button.update()
-                def change_text():
-                    return gr.Textbox("Optimization Finished! Move to the next step.", interactive = False)
 
 
     with gr.Tab(label="3 Editing"):
@@ -307,7 +300,7 @@ with gr.Blocks() as demo:
                             tgt_index = int(tgt_index)
                         )
                         run_edit_text()
-                        gr.Info('Image edit finish.')
+                        gr.Info('Image editing completed.')
                         return load_pil_img()