Update app.py

app.py

@@ -41,14 +41,13 @@ refacer = Refacer(force_cpu=args.force_cpu, colab_performance=args.colab_perform
 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:]
+def run(video_path, *vars):
+    origins = vars[:num_faces]
+    destinations = vars[num_faces:2*num_faces]
+    thresholds = vars[2*num_faces:]
 
     faces = []
-    for k in range(0, num_faces):
+    for k in range(num_faces):
         if origins[k] is not None and destinations[k] is not None:
             faces.append({
                 'origin': origins[k],
@@ -60,7 +59,7 @@ def run(*vars):
     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 the file path to show in Gradio output
+    return refaced_video_path  # Directly return the refaced video path
 
 # Prepare Gradio components
 origin = []
@@ -69,12 +68,12 @@ thresholds = []
 
 with gr.Blocks() as demo:
     with gr.Row():
-        gr.Markdown("# Refacer")
+        gr.Markdown("# Refacer - Replace Faces in Video")
     with gr.Row():
-        video = gr.Video(label="Original video", format="mp4")
-        video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
+        video = gr.Video(label="Original Video", format="mp4")
+        video2 = gr.Video(label="Refaced Video", format="mp4", interactive=False)
 
-    for i in range(0, num_faces):
+    for i in range(num_faces):
         with gr.Tab(f"Face #{i+1}"):
             with gr.Row():
                 origin.append(gr.Image(label="Face to replace"))
@@ -85,6 +84,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         button = gr.Button("Reface", variant="primary")
 
+    # Updated: Directly display the refaced video in the output
     button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
 
 # Launch the Gradio app