Update app.py

app.py

@@ -16,12 +16,10 @@ 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=512, device="cpu",side_by_side=False
+    size=img.shape[1], device="cpu",side_by_side=False
 )
 def inference(img, ver):
-    if ver == 'version 2 (🔺 robustness,🔻 stylization)':
-        out = face2paint(model2, img)
-    else:
+   
         out = face2paint(model1, img)
     return out