Update app.py

app.py

@@ -4,11 +4,26 @@ from PIL import Image
 
 os.system("wget https://www.dropbox.com/s/fgupbov77x4rrru/blendgan.pt")
 os.system("wget https://www.dropbox.com/s/v8q0dd3r4u20659/psp_encoder.pt")
+os.system("wget https://github.com/kim-ninh/align_face_ffhq/raw/main/shape_predictor_68_face_landmarks.dat")
+
+from ffhq_dataset.gen_aligned_image import FaceAlign
+
+
+fa = FaceAlign()
+import cv2
+
+
 
 def inference(content, style, index):
     content.save('content.png')
     style.save('style.png')
-    os.system("""python style_transfer_folder.py --size 1024 --add_weight_index """+str(int(index))+""" --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path style.png --input_img_path content.png""")
+    imgc = cv2.imread('content.png')
+    img_cropc = fa.get_crop_image(imgc)
+    cv2.imwrite('contentcrop.png', img_cropc)
+    imgs = cv2.imread('style.png')
+    img_crops = fa.get_crop_image(imgs)
+    cv2.imwrite('stylecrop.png', img_crops)
+    os.system("""python style_transfer_folder.py --size 1024 --add_weight_index """+str(int(index))+""" --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path stylecrop.png --input_img_path contentcrop.png""")
     return "out.jpg"
   
 title = "BlendGAN"