Ahsen Khaliq
commited on
Commit
·
3392f5b
1
Parent(s):
666c0af
Update app.py
Browse files
app.py
CHANGED
@@ -8,12 +8,12 @@ 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"
|
15 |
description = "Gradio Demo for BlendGAN: Implicitly GAN Blending for Arbitrary Stylized Face Generation. To use it, simply upload your images, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."
|
16 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2110.11728' target='_blank'>BlendGAN: Implicitly GAN Blending for Arbitrary Stylized Face Generation</a> | <a href='https://github.com/onion-liu/BlendGAN' target='_blank'>Github Repo</a></p><p style='text-align: center'>samples from repo: <img src='https://user-images.githubusercontent.com/6346064/142623312-3e6f09aa-ce88-465c-b956-a8b4db95b4da.gif' alt='animation'/> <img src='https://user-images.githubusercontent.com/6346064/142621044-086cde48-8604-467b-8c43-8768b6670ec2.gif' alt='animation'/></p>"
|
17 |
|
18 |
-
examples=[['
|
19 |
gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Image(type="pil")], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,examples=examples,allow_flagging=False).launch()
|
|
|
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"
|
15 |
description = "Gradio Demo for BlendGAN: Implicitly GAN Blending for Arbitrary Stylized Face Generation. To use it, simply upload your images, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."
|
16 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2110.11728' target='_blank'>BlendGAN: Implicitly GAN Blending for Arbitrary Stylized Face Generation</a> | <a href='https://github.com/onion-liu/BlendGAN' target='_blank'>Github Repo</a></p><p style='text-align: center'>samples from repo: <img src='https://user-images.githubusercontent.com/6346064/142623312-3e6f09aa-ce88-465c-b956-a8b4db95b4da.gif' alt='animation'/> <img src='https://user-images.githubusercontent.com/6346064/142621044-086cde48-8604-467b-8c43-8768b6670ec2.gif' alt='animation'/></p>"
|
17 |
|
18 |
+
examples=[['000000.png','100001.png']]
|
19 |
gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Image(type="pil")], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,examples=examples,allow_flagging=False).launch()
|