Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))
|
|
26 |
|
27 |
|
28 |
def face_swap(dest_img):
|
|
|
29 |
|
30 |
# Convert to RGB
|
31 |
src_img = src_img.convert(mode='RGB')
|
@@ -79,6 +80,8 @@ def greet(description,color,features,occasion,type_):
|
|
79 |
guidance_scale=7.5,
|
80 |
cross_attention_kwargs={"scale": 1.0}
|
81 |
).images[0]
|
|
|
|
|
82 |
|
83 |
return image
|
84 |
|
|
|
26 |
|
27 |
|
28 |
def face_swap(dest_img):
|
29 |
+
src_img = Image.open('./images/cat.JPG')
|
30 |
|
31 |
# Convert to RGB
|
32 |
src_img = src_img.convert(mode='RGB')
|
|
|
80 |
guidance_scale=7.5,
|
81 |
cross_attention_kwargs={"scale": 1.0}
|
82 |
).images[0]
|
83 |
+
|
84 |
+
image = face_swap(image)
|
85 |
|
86 |
return image
|
87 |
|