Upload web-ui.py

web-ui.py

@@ -174,21 +174,23 @@ with gr.Blocks() as demo:
         outputs=[output_gallery, output_text, display_seed],
     )
 	
+def start_gradio():
+    demo.launch(share=args.share, inbrowser=True)
+
 def start_ngrok():
     # Set the correct port here. By default, Gradio uses 7860, but it might be different in your setup
     gradio_port = 7860
     ngrok.set_auth_token(args.ngrok_token)
     public_url = ngrok.connect(gradio_port)
     print(f"Public URL: {public_url}")
-
-
+	
 if __name__ == "__main__":
     # Launch the Gradio app in a separate thread
-    threading.Thread(target=lambda: demo.launch(share=args.share, inbrowser=True)).start()
+    threading.Thread(target=start_gradio).start()
 
     # Give Gradio a few seconds to start up
-    time.sleep(5)
+    time.sleep(15)  # Adjust the sleep time if necessary
 
     # Start ngrok
     if args.ngrok_token:
-        start_ngrok()
+        start_ngrok()