Ahsen Khaliq
commited on
Commit
•
c58f960
1
Parent(s):
d375c2f
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,6 @@ os.system("wget https://www.dropbox.com/s/v8q0dd3r4u20659/psp_encoder.pt")
|
|
6 |
from PIL import Image
|
7 |
|
8 |
def inference(content, style):
|
9 |
-
content.resize((1024, 1024))
|
10 |
-
style.resize((1024, 1024))
|
11 |
content.save('content.png')
|
12 |
style.save('style.png')
|
13 |
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""")
|
|
|
6 |
from PIL import Image
|
7 |
|
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""")
|