Hugging Face's logo

Spaces:
akhaliq
/
AnimeGANv2
Running on T4

App Files Community
9
Ahsen Khaliq commited on
Commit
dd8ad63
•
1 Parent(s): 0d960e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,18 +3,18 @@ import torch
3
  import gradio as gr
4
 
5
  model2 = torch.hub.load(
6
- "bryandlee/animegan2-pytorch:main",
7
  "generator",
8
  pretrained=True,
9
  device="cuda",
10
  progress=False
11
  )
12
 
13
- model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
14
 
15
  face2paint = torch.hub.load(
16
- 'bryandlee/animegan2-pytorch:main', 'face2paint',
17
- size=512, device="cuda"
18
  )
19
 
20
  def inference(img, ver):
 
3
  import gradio as gr
4
 
5
  model2 = torch.hub.load(
6
+ "AK391/animegan2-pytorch:main",
7
  "generator",
8
  pretrained=True,
9
  device="cuda",
10
  progress=False
11
  )
12
 
13
+ model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
14
 
15
  face2paint = torch.hub.load(
16
+ 'AK391/animegan2-pytorch:main', 'face2paint',
17
+ size=512, device="cuda",side_by_side=False
18
  )
19
 
20
  def inference(img, ver):