Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import subprocess
 import shutil
 import os
+from PIL import Image
+import numpy as np
 
 def run_scripts(target, source, use_face_enhancer):
     if target is None or (not use_face_enhancer and source is None):
@@ -12,20 +14,21 @@ def run_scripts(target, source, use_face_enhancer):
     output_path2 = "output2" + target_extension
 
     if not use_face_enhancer:
-        # Run both scripts
         cmd1 = ["python3", "run.py", "-s", source.name, "-t", target.name, "-o", output_path1, "--frame-processor", "face_swapper"]
         subprocess.run(cmd1)
 
-    # Run the second script
     cmd2 = ["python3", "run.py", "-t", target.name if use_face_enhancer else output_path1, "-o", output_path2, "--frame-processor", "face_enhancer"]
     subprocess.run(cmd2)
 
     if not use_face_enhancer:
         os.remove(source.name)
     os.remove(target.name)
-    
-    # Return the file path so it gets displayed in the Gradio UI
-    return output_path2
+
+    # Open the image with PIL and convert to NumPy array
+    with Image.open(output_path2) as img:
+        img_array = np.array(img)
+
+    return img_array
 
 iface = gr.Interface(
     fn=run_scripts,
@@ -34,7 +37,7 @@ iface = gr.Interface(
         "file",
         gr.inputs.Checkbox(default=False, label="Use only Face Enhancer")
     ],
-    outputs="file",
+    outputs="image",  # Change the output type to image
     title="Face swapper",
     description="Upload a target image/video and a source image to swap faces.",
     live=True