Ahsen Khaliq commited on
Commit
d375c2f
1 Parent(s): c118ea5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,6 +6,8 @@ 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.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""")
@@ -16,4 +18,4 @@ description = "Gradio Demo for BlendGAN: Implicitly GAN Blending for Arbitrary S
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", shape=(1024,1024)),gr.inputs.Image(type="pil",shape=(1024,1024))], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,examples=examples,allow_flagging=False).launch()
 
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""")
 
18
  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>"
19
 
20
  examples=[['000000.png','100001.png']]
21
+ 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()