Update app.py

app.py

@@ -8,7 +8,7 @@ import requests
 model_url = "https://huggingface.co/ofter/4x-UltraSharp/resolve/main/inswapper_128.onnx"
 model_path = "./inswapper_128.onnx"
 
-# Function to download the model if it doesn't exist
+# Function to download the model
 def download_model():
     if not os.path.exists(model_path):
         print("Downloading inswapper_128.onnx...")
@@ -25,41 +25,42 @@ def download_model():
 # Download the model when the script runs
 download_model()
 
-# Argument parser for command-line options
-parser = argparse.ArgumentParser(description="Refacer")
+# 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=True, action="store_true")
-parser.add_argument("--server_name", type=str, help="Server IP address", default="0.0.0.0")
-parser.add_argument("--server_port", type=int, help="Server port", default=int(os.getenv("PORT", 7860)))
+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")
-args = parser.parse_args([])
+args = parser.parse_args()
 
 # Initialize the Refacer class
 refacer = Refacer(force_cpu=args.force_cpu, colab_performance=args.colab_performance)
+
 num_faces = args.max_num_faces
 
 # Run function for refacing video
 def run(*vars):
     video_path = vars[0]
-    origins = vars[1:(num_faces + 1)]
-    destinations = vars[(num_faces + 1):(num_faces * 2) + 1]
-    thresholds = vars[(num_faces * 2) + 1:]
+    origins = vars[1:(num_faces+1)]
+    destinations = vars[(num_faces+1):(num_faces*2)+1]
+    thresholds = vars[(num_faces*2)+1:]
 
     faces = []
     for k in range(0, num_faces):
         if origins[k] is not None and destinations[k] is not None:
             faces.append({
-                "origin": origins[k],
-                "destination": destinations[k],
-                "threshold": thresholds[k]
+                'origin': origins[k],
+                'destination': destinations[k],
+                'threshold': thresholds[k]
             })
 
     # Call refacer to process video and get file path
-    refaced_video_path = refacer.reface(video_path, faces)
+    refaced_video_path = refacer.reface(video_path, faces)  # refaced video path
     print(f"Refaced video can be found at {refaced_video_path}")
 
-    return refaced_video_path
+    return refaced_video_path  # Return the file path to show in Gradio output
 
 # Prepare Gradio components
 origin = []
@@ -74,23 +75,17 @@ with gr.Blocks() as demo:
         video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
 
     for i in range(0, num_faces):
-        with gr.Tab(f"Face #{i + 1}"):
+        with gr.Tab(f"Face #{i+1}"):
             with gr.Row():
                 origin.append(gr.Image(label="Face to replace"))
                 destination.append(gr.Image(label="Destination face"))
             with gr.Row():
                 thresholds.append(gr.Slider(label="Threshold", minimum=0.0, maximum=1.0, value=0.2))
-
+    
     with gr.Row():
         button = gr.Button("Reface", variant="primary")
 
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
-# Launch the Gradio app with optional share functionality
-demo.queue().launch(
-    show_error=True,
-    share=args.share_gradio,  # Share option for public access
-    server_name=args.server_name,  # Server IP address
-    server_port=args.server_port,  # Port number
-    debug=True  # Enable debug mode for better error handling
-)
+# Launch the Gradio app
+demo.queue().launch(show_error=True, share=args.share_gradio, server_name="0.0.0.0", server_port=args.server_port)