Update app.py

app.py

@@ -14,13 +14,23 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
 
         client = Client("tuan2308/face-swap")
 
+        source_filename = f"source_{random.randint(1000, 9999)}.jpg"
+        target_filename = f"target_{random.randint(1000, 9999)}.jpg"
+
+        source_img.save(source_filename)
+        target_img.save(target_filename)
+
         result = client.predict(
-            source_file=handle_file(source_img),
-            target_file=handle_file(target_img),
+            source_file=handle_file(source_filename),
+            target_file=handle_file(target_filename),
             doFaceEnhancer=doFaceEnhancer,
             api_name="/predict",
             api_key=api_key
         )
+
+        os.remove(source_filename)
+        os.remove(target_filename)
+
         return result
     except Exception as e:
         print(f"Ошибка при вызове API: {e}")
@@ -29,11 +39,11 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
 iface = gr.Interface(
     fn=swap_face_api,
     inputs=[
-        gr.Image(type="filepath", label="Source Image"),
-        gr.Image(type="filepath", label="Target Image"),
+        gr.Image(type="pil", label="Source Image"),
+        gr.Image(type="pil", label="Target Image"),
         gr.Checkbox(label="Face Enhancer?")
     ],
-    outputs=gr.Image(label="Output Image"),
+    outputs=gr.Image(type="pil", label="Output Image"),
     title="Face Swap via API"
 )