Add CPU support

app.py

@@ -51,8 +51,11 @@ def save_merged_boxes(predictions, image_np):
     return roi
 
 # Load the EfficientNet model
-def load_model(model_path):
-    model = torch.load(model_path)
+def load_model(model_path, map_location=None):
+    if map_location:
+        model = torch.load(model_path, map_location=map_location)
+    else:
+        model = torch.load(model_path)
     model = model.to(device)
     model.eval()  # Set the model to evaluation mode
     return model