Update app.py

app.py

@@ -9,7 +9,8 @@ import torchvision.transforms as transforms
 CLASSES = ['guro', 'pigs', 'proofs', 'protyk', 'safe', 'shit']
 NUM_CLASSES = len(CLASSES)
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+#device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = 'cpu'
 
 model = models.resnet18(pretrained=True)
 model.fc = nn.Linear(model.fc.in_features, NUM_CLASSES)