Update app.py

app.py

@@ -49,7 +49,7 @@ model = Cnn()
 model_weights_path = 'model_weights.pth'
 
 # Load the model weights
-model.load_state_dict(torch.load(model_weights_path))
+model.load_state_dict(torch.load(model_weights_path),map_location=torch.device('cpu') )
 
 # Set the model to evaluation mode for inference
 model.eval()