Update app.py

app.py

@@ -24,7 +24,7 @@ def inference(content, style, index):
     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""")
+    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""")
     return "out.jpg"
   
 title = "BlendGAN"