Ahsen Khaliq commited on
Commit
6dcb58e
1 Parent(s): 416c8ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -21,9 +21,6 @@ def inference(content, style, index):
21
  imgc = cv2.imread('content.png')
22
  img_cropc = fa.get_crop_image(imgc)
23
  cv2.imwrite('contentcrop.png', img_cropc)
24
- imgs = cv2.imread('style.png')
25
- img_crops = fa.get_crop_image(imgs)
26
- cv2.imwrite('stylecrop.png', img_crops)
27
  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 contentcrop.png""")
28
  return "out.jpg"
29
 
21
  imgc = cv2.imread('content.png')
22
  img_cropc = fa.get_crop_image(imgc)
23
  cv2.imwrite('contentcrop.png', img_cropc)
 
 
 
24
  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 contentcrop.png""")
25
  return "out.jpg"
26