Ahsen Khaliq commited on
Commit
07c3194
1 Parent(s): 147df81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  from PIL import Image
3
  import torch
4
  import gradio as gr
5
-
6
  import torch
7
  torch.backends.cudnn.benchmark = True
8
  from torchvision import transforms, utils
@@ -148,4 +148,4 @@ description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, si
148
  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>"
149
 
150
  examples=[['mona.png','Jinx']]
151
- gr.Interface(inference, [gr.inputs.Image(type="filepath"),gr.inputs.Dropdown(choices=['JoJo', 'Disney','Jinx','Caitlyn','Yasuho','Arcane Multi','Art','Spider-Verse'], type="value", default='JoJo', label="Model")], gr.outputs.Image(type="file"),title=title,description=description,article=article,allow_flagging="never",examples=examples,allow_screenshot=False).launch(enable_queue=True,cache_examples=False)
 
2
  from PIL import Image
3
  import torch
4
  import gradio as gr
5
+ os.system("pip install gradio==2.5.3")
6
  import torch
7
  torch.backends.cudnn.benchmark = True
8
  from torchvision import transforms, utils
 
148
  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>"
149
 
150
  examples=[['mona.png','Jinx']]
151
+ gr.Interface(inference, [gr.inputs.Image(type="filepath"),gr.inputs.Dropdown(choices=['JoJo', 'Disney','Jinx','Caitlyn','Yasuho','Arcane Multi','Art','Spider-Verse'], type="value", default='JoJo', label="Model")], gr.outputs.Image(type="file"),title=title,description=description,article=article,allow_flagging="never",examples=examples,allow_screenshot=False,enable_queue=True).launch()