Update app.py

app.py

@@ -208,13 +208,8 @@ function custom(){
 }
 """
 
-if __name__ == "__main__":
-    
-    port_inc()
-    _dynamo.config.suppress_errors = True
-    mp.set_start_method("spawn", force=True)
+def ui():
     result = []
-    
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Column(elem_id="col-container"):
             gr.Markdown(f"""
@@ -242,4 +237,10 @@ if __name__ == "__main__":
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
         run_button.click(fn=main,inputs=[prompt,prompt2],outputs=result)
-        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
+        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
+
+mp.set_start_method("spawn", force=True)
+ui()
+if __name__ == "__main__":
+    port_inc()
+    _dynamo.config.suppress_errors = True