update: app.py

app.py

@@ -341,7 +341,7 @@ def process_video(
     return targets, t_facial_masks, Ms, original_frames, names, fps
 
 
-def swap_image_gr(img1, img2, use_post=False, use_gpen=False, gpu_mode=True):
+def swap_image_gr(img1, img2, use_post=False, use_gpen=False, ):
     root_dir = make_abs_path("./online_data")
     req_id = uuid.uuid1().hex
     data_dir = os.path.join(root_dir, req_id)
@@ -369,7 +369,7 @@ def swap_image_gr(img1, img2, use_post=False, use_gpen=False, gpu_mode=True):
     return out
 
 
-def swap_video_gr(img1, target_path, use_gpu=True, frames=9999999):
+def swap_video_gr(img1, target_path, frames=9999999):
     root_dir = make_abs_path("./online_data")
     req_id = uuid.uuid1().hex
     data_dir = os.path.join(root_dir, req_id)
@@ -460,12 +460,12 @@ if __name__ == "__main__":
                     video_button = gr.Button("换脸")
         image_button.click(
             swap_image_gr,
-            inputs=[image1_input, image2_input, use_post, use_gpen, use_gpu],
+            inputs=[image1_input, image2_input, use_post, use_gpen],
             outputs=image_output,
         )
         video_button.click(
             swap_video_gr,
-            inputs=[image3_input, video_input, use_gpu],
+            inputs=[image3_input, video_input],
             outputs=video_output,
         )