Update app.py

app.py

@@ -127,13 +127,9 @@ with gr.Blocks(title="Astro-Diffusion: Base vs LoRA") as demo:
     )
 
 
-port = int(os.getenv("GRADIO_SERVER_PORT", "7861"))
-share = os.getenv("GRADIO_PUBLIC_SHARE", "False") == "True"
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=port,
-    show_error=True,
-    share=share,
-)
-
+if __name__ == "__main__":
+    interface = build_ui()
+    port = int(os.getenv("PORT", os.getenv("GRADIO_SERVER_PORT", "7861")))
+    share = os.getenv("GRADIO_PUBLIC_SHARE", "True").lower() == "true"
+    interface.launch(server_name="0.0.0.0", server_port=port, share=share)