Update app.py

app.py

@@ -90,7 +90,7 @@ def predict_image(image, model):
 
 # Load the EfficientNet model outside the function to avoid loading it multiple times
 model_path = 'vehicle.pt'
-model = load_model(model_path)
+model = load_model(model_path, map_location=torch.device('cpu'))
 
 # Gradio UI
 title = "Vehicle Collision Classification"