change files

app.py

@@ -7,7 +7,7 @@ import requests
 
 # Load the pre-trained MobileNetV2 model from torchvision
 model = models.mobilenet_v2(pretrained=True)
-device = torch.device("mps")
+device = torch.device("cpu")
 model.to(device)
 model.eval()  # Set model to evaluation mode
 
@@ -33,7 +33,7 @@ def preprocess_image(image):
 # Prediction function
 def predict(image):
     # Load the input file
-    reloaded_img_tensor = torch.load(image).to(torch.device("mps"))
+    reloaded_img_tensor = torch.load(image).to(torch.device("cpu"))
     
     # Make predictions
     output = model(reloaded_img_tensor)