Update app.py

app.py

@@ -25,8 +25,8 @@ app = FaceAnalysis(name='buffalo_l')
 app.prepare(ctx_id=0, det_size=(640, 640))
 
 
-def face_swap(dest_img):
-    src_img = Image.open('./images/cat.JPG')
+def face_swap(src_img, dest_img):
+    src_img = Image.open('./images/' + src_img + '.JPG')
     
     # Convert to RGB
     src_img = src_img.convert(mode='RGB')
@@ -77,7 +77,8 @@ def greet(description,color,features,occasion,type_,face):
         cross_attention_kwargs={"scale": 1.0}
         ).images[0]
 
-    image = face_swap(image)
+    if(face != 'Normal'):
+        image = face_swap(face, image)
     
     return image