Ahsen Khaliq commited on
Commit
085c620
1 Parent(s): 3392f5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  def inference(content, style):
9
  content.save('content.png')
10
  style.save('style.png')
11
- os.system("""python style_transfer_folder.py --size 512 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path style.png --input_img_path content.png""")
12
  return "out.jpg"
13
 
14
  title = "BlendGAN"
 
8
  def inference(content, style):
9
  content.save('content.png')
10
  style.save('style.png')
11
+ os.system("""python style_transfer_folder.py --size 1024 --ckpt ./blendgan.pt --psp_encoder_ckpt ./psp_encoder.pt --style_img_path style.png --input_img_path content.png""")
12
  return "out.jpg"
13
 
14
  title = "BlendGAN"