yy

app.py

@@ -1,5 +1,6 @@
 import os
 import gradio as gr
+from PIL import Image
 from roop.core import (
     decode_execution_providers,
     suggest_max_memory,
@@ -13,7 +14,7 @@ from roop.utilities import normalize_output_path
 app = gr.Blocks()
 
 with app:
-   
+    
     source_image = gr.Image(label="Source Image")
     target_images = gr.File(label="Target Images", file_count="multiple")
     do_face_enhancer = gr.Checkbox(label="Face Enhancer")
@@ -27,8 +28,15 @@ with app:
         for target_image in target_images:
             source_path = "input.jpg"
             target_path = "target.jpg"
-            source_image.save(source_path)
-            target_image.save(target_path)
+
+            # Convert gr.Image to PIL Image
+            source_pil_image = Image.fromarray(source_image)
+            source_pil_image.save(source_path)
+
+            # Convert gr.File to PIL Image
+            target_pil_image = Image.open(target_image.name)
+            target_pil_image.save(target_path)
+
             output_path = "output.jpg"
             roop.globals.source_path = source_path
             roop.globals.target_path = target_path
@@ -37,14 +45,25 @@ with app:
                 roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
             else:
                 roop.globals.frame_processors = ["face_swapper"]
-            roop.globals.head_fps = True
+            roop.globals.headless = True
+            roop.globals.keep_fps = True
             roop.globals.keep_audio = True
             roop.globals.keep_frames = False
-            roop.globals.video_codec = "libx264"
+            roop.globals.many_faces = False
+            roop.globals.video_encoder = "libx264"
+            roop.globals.video_quality = 18
+            roop.globals.max_memory = suggest_max_memory()
             roop.globals.execution_providers = decode_execution_providers(["cuda"])
             roop.globals.execution_threads = suggest_execution_threads()
-            swap.swap_faces()
-            result_image = roop.globals.output_path
+
+            for frame_processor in get_frame_processors_modules(
+                roop.globals.frame_processors
+            ):
+                if not frame_processor.pre_check():
+                    return
+
+            swap.start()
+            result_image = Image.open(roop.globals.output_path)
             result_images.append(result_image)
         return result_images