Ahsen Khaliq commited on
Commit
46afa05
1 Parent(s): 38887ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
3
  import torch
4
  import gradio as gr
5
 
 
6
  os.system("pip install dlib")
7
  import torch
8
  torch.backends.cudnn.benchmark = True
@@ -88,4 +89,4 @@ description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, si
88
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.11641' target='_blank'>JoJoGAN: One Shot Face Stylization</a>| <a href='https://github.com/mchong6/JoJoGAN' target='_blank'>Github Repo Pytorch</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_jojogan' alt='visitor badge'></center> <p style='text-align: center'>samples from repo: <img src='https://raw.githubusercontent.com/mchong6/JoJoGAN/main/teaser.jpg' alt='animation'/></p>"
89
 
90
  examples=[['iu.jpeg']]
91
- gr.Interface(inference, [gr.inputs.Image(type="filepath",shape=(512,512))], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,allow_flagging=False,examples=examples).launch()
 
3
  import torch
4
  import gradio as gr
5
 
6
+
7
  os.system("pip install dlib")
8
  import torch
9
  torch.backends.cudnn.benchmark = True
 
89
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.11641' target='_blank'>JoJoGAN: One Shot Face Stylization</a>| <a href='https://github.com/mchong6/JoJoGAN' target='_blank'>Github Repo Pytorch</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_jojogan' alt='visitor badge'></center> <p style='text-align: center'>samples from repo: <img src='https://raw.githubusercontent.com/mchong6/JoJoGAN/main/teaser.jpg' alt='animation'/></p>"
90
 
91
  examples=[['iu.jpeg']]
92
+ gr.Interface(inference, [gr.inputs.Image(type="filepath",shape=(256,256))], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,allow_flagging=False,examples=examples).launch()