Update app.py

app.py

@@ -1,27 +1,16 @@
 import gradio as gr
 from refacer import Refacer
-import argparse
 import ngrok
 import time
-
-# Argument parser
-parser = argparse.ArgumentParser(description='Refacer')
-parser.add_argument("--max_num_faces", type=int, help="Max number of faces on UI", default=5)
-parser.add_argument("--force_cpu", help="Force CPU mode", default=False, action="store_true")
-parser.add_argument("--share_gradio", help="Share Gradio", default=False, action="store_true")
-parser.add_argument("--server_name", type=str, help="Server IP address", default="127.0.0.1")
-parser.add_argument("--server_port", type=int, help="Server port", default=7860)
-parser.add_argument("--colab_performance", help="Use in colab for better performance", default=False, action="store_true")
-parser.add_argument("--ngrok", type=str, help="Use ngrok", default=None)
-parser.add_argument("--ngrok_region", type=str, help="ngrok region", default="us")
-args = parser.parse_args()
+import threading
 
 # Initialize Refacer
-print("Initializing Refacer...")
-start_time = time.time()
-refacer = Refacer(force_cpu=args.force_cpu, colab_performance=args.colab_performance)
-num_faces = args.max_num_faces
-print(f"Refacer initialized in {time.time() - start_time:.2f} seconds")
+def initialize_refacer():
+    global refacer
+    print("Initializing Refacer...")
+    start_time = time.time()
+    refacer = Refacer(force_cpu=False, colab_performance=False)
+    print(f"Refacer initialized in {time.time() - start_time:.2f} seconds")
 
 # Ngrok connection
 def connect(token, port, options):
@@ -65,6 +54,7 @@ def run(*vars):
 origin = []
 destination = []
 thresholds = []
+num_faces = 5
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -86,8 +76,9 @@ with gr.Blocks() as demo:
 
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
-if args.ngrok:
-    connect(args.ngrok, args.server_port, {'region': args.ngrok_region, 'authtoken_from_env': False})
+# Start initialization in a separate thread to prevent blocking
+init_thread = threading.Thread(target=initialize_refacer)
+init_thread.start()
 
 # Launch demo
-demo.queue().launch(show_error=True, share=args.share_gradio, server_name=args.server_name, server_port=args.server_port)
+demo.queue().launch(show_error=True, share=True, server_name="0.0.0.0", server_port=7860)