Update app.py

app.py

@@ -7,7 +7,7 @@ from torchvision import transforms
 
 # Load model pretrained
 try:
-    model = torch.load("./abudhabi.pt")
+    model = torch.load("./abudhabi.pt", map_location=torch.device('cpu'))
     model.eval()
     print("Model loaded successfully!")
 except Exception as e: