Update app.py

app.py

@@ -244,7 +244,8 @@ with gr.Blocks() as demo:
                         run_optimization()
                         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:
+                    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)