Update app.py

app.py

@@ -14,12 +14,14 @@ swapper = insightface.model_zoo.get_model('./inswapper_128.onnx', download=False
 
 def generate_image(src_img, dest_img):
     # Convert to RGB
-    src_img = src_img.convert(mode='RGB')
-    dest_img = dest_img.convert(mode='RGB')
+    #src_img = src_img.convert(mode='RGB')
+    #dest_img = dest_img.convert(mode='RGB')
 
     # Convert to array
-    src_img_arr = np.asarray(src_img)
-    dest_img_arr = np.asarray(dest_img)
+    #src_img_arr = np.asarray(src_img)
+    #dest_img_arr = np.asarray(dest_img)
+    src_img_arr = src_img
+    dest_img_arr = dest_img
 
     # Face detection
     src_faces = app.get(src_img_arr)