avinev2_04

Runtime error

adpro commited on Jan 30, 2023

Commit

4337bf7

1 Parent(s): 606366d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ model2 = torch.hub.load(
 )
-model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v2",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
     size=512, device="cpu",side_by_side=False

 )
+model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="paprika",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
     size=512, device="cpu",side_by_side=False