Update app.py
Browse files
app.py
CHANGED
@@ -14,13 +14,23 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
|
|
14 |
|
15 |
client = Client("tuan2308/face-swap")
|
16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
result = client.predict(
|
18 |
-
source_file=handle_file(
|
19 |
-
target_file=handle_file(
|
20 |
doFaceEnhancer=doFaceEnhancer,
|
21 |
api_name="/predict",
|
22 |
api_key=api_key
|
23 |
)
|
|
|
|
|
|
|
|
|
24 |
return result
|
25 |
except Exception as e:
|
26 |
print(f"Ошибка при вызове API: {e}")
|
@@ -29,11 +39,11 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
|
|
29 |
iface = gr.Interface(
|
30 |
fn=swap_face_api,
|
31 |
inputs=[
|
32 |
-
gr.Image(type="
|
33 |
-
gr.Image(type="
|
34 |
gr.Checkbox(label="Face Enhancer?")
|
35 |
],
|
36 |
-
outputs=gr.Image(label="Output Image"),
|
37 |
title="Face Swap via API"
|
38 |
)
|
39 |
|
|
|
14 |
|
15 |
client = Client("tuan2308/face-swap")
|
16 |
|
17 |
+
source_filename = f"source_{random.randint(1000, 9999)}.jpg"
|
18 |
+
target_filename = f"target_{random.randint(1000, 9999)}.jpg"
|
19 |
+
|
20 |
+
source_img.save(source_filename)
|
21 |
+
target_img.save(target_filename)
|
22 |
+
|
23 |
result = client.predict(
|
24 |
+
source_file=handle_file(source_filename),
|
25 |
+
target_file=handle_file(target_filename),
|
26 |
doFaceEnhancer=doFaceEnhancer,
|
27 |
api_name="/predict",
|
28 |
api_key=api_key
|
29 |
)
|
30 |
+
|
31 |
+
os.remove(source_filename)
|
32 |
+
os.remove(target_filename)
|
33 |
+
|
34 |
return result
|
35 |
except Exception as e:
|
36 |
print(f"Ошибка при вызове API: {e}")
|
|
|
39 |
iface = gr.Interface(
|
40 |
fn=swap_face_api,
|
41 |
inputs=[
|
42 |
+
gr.Image(type="pil", label="Source Image"),
|
43 |
+
gr.Image(type="pil", label="Target Image"),
|
44 |
gr.Checkbox(label="Face Enhancer?")
|
45 |
],
|
46 |
+
outputs=gr.Image(type="pil", label="Output Image"),
|
47 |
title="Face Swap via API"
|
48 |
)
|
49 |
|