Update app.py

app.py

@@ -245,9 +245,8 @@ with gr.Blocks() as demo:
                         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 Exception as e:
-                        print(e)
-                        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)
+                        gr.Error("e")
+                        return "Error: use a smaller batch size or try latter.",gr.Button("Run Editing (Check log for progress.)",interactive = False)