Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 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):
@@ -43,9 +45,9 @@ def main():
 
         if st.button("Swap Faces"):
             with st.spinner("Swapping Faces..."):
-                output_path = swap_face(
-                    np.array(source_image),
-                    np.array(target_image),
+                output_path = run_scripts(
+                    target_file,
+                    source_file if not doFaceEnhancer else None,
                     doFaceEnhancer,
                 )
                 output_image = Image.open(output_path)