Ahsen Khaliq
commited on
Commit
·
085c620
1
Parent(s):
3392f5b
Update app.py
Browse files
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
|
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"
|