Aiconvert commited on
Commit
bf70e48
1 Parent(s): a3ce083

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -51
app.py CHANGED
@@ -1,53 +1,4 @@
1
- import gradio as gr
2
- import insightface
3
- from insightface.app import FaceAnalysis
4
 
5
- assert insightface.__version__ >= '0.7'
6
-
7
- def prepare_app():
8
- app = FaceAnalysis(name='buffalo_l')
9
- app.prepare(ctx_id=0, det_size=(640, 640))
10
- swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True, download_zip=True)
11
- return app, swapper
12
-
13
- def swap_faces(sourceImage, destinationImage):
14
- """Swaps faces between the source and destination images."""
15
- app, swapper = prepare_app()
16
-
17
- source_faces = app.get(sourceImage)
18
- if not source_faces:
19
- raise gr.Error("No face found in the source image.")
20
- source_face = source_faces[0]
21
-
22
- destination_faces = app.get(destinationImage)
23
- if not destination_faces:
24
- raise gr.Error("No face found in the destination image.")
25
- destination_face = destination_faces[0]
26
-
27
- result = swapper.get(destinationImage, destination_face, source_face, paste_back=True)
28
- return result
29
-
30
-
31
- gr.Interface(
32
-
33
- swap_faces,
34
- [
35
- gr.Image(label="Source Image"),
36
- gr.Image(label="Destination Image"),
37
- ],
38
- gr.Image(),
39
- examples=[
40
- ['./examples/IMG-a.jpg', './examples/IMG-a1.jpg'],
41
- ['./examples/IMG-b.jpg', './examples/IMG-b1.jpg'],
42
- ['./examples/IMG-c.jpg', './examples/IMG-c1.jpg'],
43
- ['./examples/IMG-d.jpg', './examples/IMG-d1.jpg'],
44
- ['./examples/IMG-a.jpg', './examples/IMG-a2.jpg'],
45
- ['./examples/IMG-b1.jpg', './examples/IMG-b11.jpg'],
46
- ],
47
- theme="syddharth/gray-minimal",
48
- title="<h1 style='color: black; text-decoration: underline;'>AI Face Swapping - Free Face Swap</h1>",
49
- description="<span>● Experience the power of Ai Reface with Mergify.</span><br><span>● Swap faces with anyone in seconds - photos or AI images!</span><br><span>● A gradio demo for <a href='https://aiconvert.online/ai-face-merge/' style='color: blue; text-decoration: none;'>Mergify's Ai Face Swap </a></span><br><span>● Full features and Multiple Face Swapping For Free with no limitations at <a href='https://aiconvert.online' style='color: crimson; text-decoration: underline;'>Aiconvert.online</a></span>",
50
-
51
- css=" footer{display:none !important;}"
52
- ).launch()
53
 
 
 
1
+ import os
 
 
2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
 
4
+ exec(os.environ.get('CODE'))