Rooni commited on
Commit
22a4581
·
verified ·
1 Parent(s): 570f004

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
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(source_img),
19
- target_file=handle_file(target_img),
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="filepath", label="Source Image"),
33
- gr.Image(type="filepath", label="Target Image"),
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