Ahsen Khaliq
commited on
Commit
·
59a2d8d
1
Parent(s):
e29e9b5
Update app.py
Browse files
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"
|