Hugging Face's logo

Spaces:
akhaliq
/
AnimeGANv2
Running on T4

App Files Community
9
Ahsen Khaliq commited on
Commit
45f8f81
•
1 Parent(s): 11f9073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ model2 = torch.hub.load(
16
  model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
17
  face2paint = torch.hub.load(
18
  'AK391/animegan2-pytorch:main', 'face2paint',
19
- size=256, device="cuda",side_by_side=False
20
  )
21
  def inference(img, ver):
22
  if ver == 'version 2 (🔺 robustness,🔻 stylization)':
16
  model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
17
  face2paint = torch.hub.load(
18
  'AK391/animegan2-pytorch:main', 'face2paint',
19
+ size=512, device="cuda",side_by_side=False
20
  )
21
  def inference(img, ver):
22
  if ver == 'version 2 (🔺 robustness,🔻 stylization)':