Hugging Face's logo

Spaces:
sudhir2016
/
Anime
Sleeping

App Files Community
sudhir2016 commited on
Commit
b892439
·
1 Parent(s): d4580e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model = torch.hub.load(
6
  "generator",
7
  pretrained=True)
8
  face2paint = torch.hub.load(
9
- 'AK391/animegan2-pytorch:main', 'face2paint')
10
  def inference(img):
11
  out = face2paint(model, img)
12
  return out
 
6
  "generator",
7
  pretrained=True)
8
  face2paint = torch.hub.load(
9
+ 'AK391/animegan2-pytorch:main', 'face2paint',trust_repo=True,size=512)
10
  def inference(img):
11
  out = face2paint(model, img)
12
  return out