cpu error

app.py

@@ -16,7 +16,7 @@ label_encoder_type = joblib.load(LABEL_ENCODER_TYPE_PATH)
 
 # Load the model
 model = MultiOutputModel(num_classes_type=len(label_encoder_type.classes_))
-model.load_state_dict(torch.load(MODEL_PATH))
+model.load_state_dict(torch.load(MODEL_PATH,map_location=torch.device('cpu')))
 model.eval()  # Set the model to evaluation mode
 
 # Define the image transformations