Upload web-ui.py

web-ui.py

@@ -10,12 +10,15 @@ from ip_adapter.ip_adapter_faceid import IPAdapterFaceIDPlus
 import argparse
 import random
 from insightface.utils import face_align
+from pyngrok import ngrok
 
 # Argument parser for command line options
 parser = argparse.ArgumentParser()
 parser.add_argument("--share", action="store_true", help="Enable Gradio share option")
 parser.add_argument("--num_images", type=int, default=1, help="Number of images to generate")
 parser.add_argument("--cache_limit", type=int, default=1, help="Limit for model cache")
+parser.add_argument("--ngrok_token", type=str, default=None, help="ngrok authtoken for tunneling")
+
 args = parser.parse_args()
 
 # Add new model names here
@@ -168,5 +171,14 @@ with gr.Blocks() as demo:
         inputs=[input_image, positive_prompt, negative_prompt, width, height, model_selector, num_inference_steps, seed, randomize_seed, num_images, batch_size, enable_shortcut, s_scale],
         outputs=[output_gallery, output_text, display_seed],
     )
-
-demo.launch(share=args.share, inbrowser=True)
+	
+def start_ngrok():
+    if args.ngrok_token:
+        ngrok.set_auth_token(args.ngrok_token)
+        url = ngrok.connect(port=7860)  # You can choose your desired port
+        print(f"ngrok tunnel started at {url}")	
+
+if __name__ == "__main__":
+    if args.ngrok_token:
+        start_ngrok()
+    demo.launch(share=args.share, inbrowser=True)