avinev2_01

Runtime error

adpro commited on Jan 30, 2023

Commit

20bd19f

1 Parent(s): 9229411

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model2 = torch.hub.load(
 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="celeba_distill",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
-    size=img.shape[1], device="cpu",side_by_side=False
 )
 def inference(img, ver):

 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="celeba_distill",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
+    size=1024, device="cpu",side_by_side=False
 )
 def inference(img, ver):