Update app.py

app.py

@@ -10,14 +10,14 @@ from PIL import Image
 from torchvision.transforms.functional import center_crop, to_tensor
 
 
-
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 print("🧠 Loading Model...")
 model = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cuda",
+    device=device,
     progress=True,
 )