echolee commited on
Commit
addcdcf
1 Parent(s): 88e15ba

clear info

Browse files
Files changed (1) hide show
  1. app.py +15 -9
app.py CHANGED
@@ -9,21 +9,27 @@ model2 = torch.hub.load(
9
  device="cpu",
10
  progress=False
11
  )
12
- model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cpu")
 
13
  face2paint = torch.hub.load(
14
- 'AK391/animegan2-pytorch:main', 'face2paint',
15
- size=512, device="cpu",side_by_side=False
16
  )
 
 
17
  def inference(img, ver):
18
  if ver == 'version 2 (🔺 robustness,🔻 stylization)':
19
  out = face2paint(model2, img)
20
  else:
21
  out = face2paint(model1, img)
22
  return out
23
-
24
- title = "AnimeGANv2"
 
25
  description = "Gradio Demo for AnimeGanv2 Face Portrait v2. To use it, simply upload your image, 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."
26
- article = "<p style='text-align: center'><a href='https://github.com/bryandlee/animegan2-pytorch' target='_blank'>Github Repo Pytorch</a> | <a href='https://github.com/Kazuhito00/AnimeGANv2-ONNX-Sample' target='_blank'>Github Repo ONNX</a></p><p style='text-align: center'>samples from repo: <img src='https://user-images.githubusercontent.com/26464535/129888683-98bb6283-7bb8-4d1a-a04a-e795f5858dcf.gif' alt='animation'/> <img src='https://user-images.githubusercontent.com/26464535/137619176-59620b59-4e20-4d98-9559-a424f86b7f24.jpg' alt='animation'/></p>"
27
- examples=[['groot.jpeg','version 2 (🔺 robustness,🔻 stylization)'],['bill.png','version 1 (🔺 stylization, 🔻 robustness)'],['tony.png','version 1 (🔺 stylization, 🔻 robustness)'],['elon.png','version 2 (🔺 robustness,🔻 stylization)'],['IU.png','version 1 (🔺 stylization, 🔻 robustness)'],['billie.png','version 2 (🔺 robustness,🔻 stylization)'],['will.png','version 2 (🔺 robustness,🔻 stylization)'],['beyonce.jpeg','version 1 (🔺 stylization, 🔻 robustness)'],['gongyoo.jpeg','version 1 (🔺 stylization, 🔻 robustness)']]
28
- gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Radio(['version 1 (🔺 stylization, 🔻 robustness)','version 2 (🔺 robustness,🔻 stylization)'], type="value", default='version 2 (🔺 robustness,🔻 stylization)', label='version')
29
- ], gr.outputs.Image(type="pil"),title=title,description=description,article=article,enable_queue=True,examples=examples,allow_flagging=False).launch()
 
 
 
9
  device="cpu",
10
  progress=False
11
  )
12
+ model1 = torch.hub.load("AK391/animegan2-pytorch:main",
13
+ "generator", pretrained="face_paint_512_v1", device="cpu")
14
  face2paint = torch.hub.load(
15
+ 'AK391/animegan2-pytorch:main', 'face2paint',
16
+ size=512, device="cpu", side_by_side=False
17
  )
18
+
19
+
20
  def inference(img, ver):
21
  if ver == 'version 2 (🔺 robustness,🔻 stylization)':
22
  out = face2paint(model2, img)
23
  else:
24
  out = face2paint(model1, img)
25
  return out
26
+
27
+
28
+ title = "Face Trans For You"
29
  description = "Gradio Demo for AnimeGanv2 Face Portrait v2. To use it, simply upload your image, 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."
30
+ # article = "<p style='text-align: center'><a href='https://github.com/bryandlee/animegan2-pytorch' target='_blank'>Github Repo Pytorch</a> | <a href='https://github.com/Kazuhito00/AnimeGANv2-ONNX-Sample' target='_blank'>Github Repo ONNX</a></p><p style='text-align: center'>samples from repo: <img src='https://user-images.githubusercontent.com/26464535/129888683-98bb6283-7bb8-4d1a-a04a-e795f5858dcf.gif' alt='animation'/> <img src='https://user-images.githubusercontent.com/26464535/137619176-59620b59-4e20-4d98-9559-a424f86b7f24.jpg' alt='animation'/></p>"
31
+ article = "<p style='text-align: center'>samples: <img src='https://user-images.githubusercontent.com/26464535/129888683-98bb6283-7bb8-4d1a-a04a-e795f5858dcf.gif' alt='animation'/> <img src='https://user-images.githubusercontent.com/26464535/137619176-59620b59-4e20-4d98-9559-a424f86b7f24.jpg' alt='animation'/></p>"
32
+ examples = [['groot.jpeg', 'version 2 (🔺 robustness,🔻 stylization)'], ['bill.png', 'version 1 (🔺 stylization, 🔻 robustness)'], ['tony.png', 'version 1 (🔺 stylization, 🔻 robustness)'], ['elon.png', 'version 2 (🔺 robustness,🔻 stylization)'], ['IU.png', 'version 1 (🔺 stylization, 🔻 robustness)'], [
33
+ 'billie.png', 'version 2 (🔺 robustness,🔻 stylization)'], ['will.png', 'version 2 (🔺 robustness,🔻 stylization)'], ['beyonce.jpeg', 'version 1 (🔺 stylization, 🔻 robustness)'], ['gongyoo.jpeg', 'version 1 (🔺 stylization, 🔻 robustness)']]
34
+ gr.Interface(inference, [gr.inputs.Image(type="pil"), gr.inputs.Radio(['version 1 (🔺 stylization, 🔻 robustness)', 'version 2 (🔺 robustness,🔻 stylization)'], type="value", default='version 2 (🔺 robustness,🔻 stylization)', label='version')
35
+ ], gr.outputs.Image(type="pil"), title=title, description=description, article=article, enable_queue=True, examples=examples, allow_flagging=False).launch()