Ahsen Khaliq commited on
Commit
59a2d8d
·
1 Parent(s): e29e9b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ os.system("wget https://www.dropbox.com/s/v8q0dd3r4u20659/psp_encoder.pt")
8
  def inference(content, style, index):
9
  content.save('content.png')
10
  style.save('style.png')
11
- os.system("""python style_transfer_folder.py --size 1024 --add_weight_index """+int(index)+""" --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, index):
9
  content.save('content.png')
10
  style.save('style.png')
11
+ 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""")
12
  return "out.jpg"
13
 
14
  title = "BlendGAN"