Update app.py

app.py

@@ -7,7 +7,7 @@ from model import ExampleModel
 model_path = "animal_7.pth"
 labels = ["bird", "cat", "dog", "horse"]
 num_classes = len(labels)
-device = torch.device("cuda:0" if torch.cuda.is_available() else "mps")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")
 
 
 # Preprocess