Update app.py

app.py

@@ -8,15 +8,15 @@ model2 = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cuda",
+    device="cpu",
     progress=False
 )
 
 
-model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cuda")
+model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint', 
-    size=1024, device="cuda",side_by_side=False
+    size=1024, device="cpu",side_by_side=False
 )
 def inference(img, ver):
     if ver == 'version 2 (🔺 robustness,🔻 stylization)':