tonyassi commited on
Commit
10ef849
1 Parent(s): 9b22d92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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