Update app.py

app.py

@@ -29,14 +29,13 @@ download_model()
 refacer = Refacer(force_cpu=False, colab_performance=False)
 
 # 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[:5]
+    destinations = vars[5:10]
+    thresholds = vars[10:]
 
     faces = []
-    for k in range(0, num_faces):
+    for k in range(5):
         if origins[k] is not None and destinations[k] is not None:
             faces.append({
                 'origin': origins[k],
@@ -59,11 +58,6 @@ def run(*vars):
     return video_buffer  # Gradio will handle the video display
 
 # Prepare Gradio components
-num_faces = 5
-origin = []
-destination = []
-thresholds = []
-
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("# Refacer")
@@ -71,19 +65,19 @@ with gr.Blocks() as demo:
         video = gr.Video(label="Original video", format="mp4")
         video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
 
-    for i in range(0, num_faces):
+    origins, destinations, thresholds = [], [], []
+    for i in range(5):
         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"))
+                origins.append(gr.Image(label="Face to replace"))
+                destinations.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")
 
-    # Click event: Refacing the video and showing the refaced video in Gradio
-    button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
+    button.click(fn=run, inputs=[video] + origins + destinations + thresholds, outputs=[video2])
 
 # Launch the Gradio app
 demo.queue().launch(show_error=True, server_name="0.0.0.0", server_port=7860)