diff --git a/bird_cnn/__pycache__/server.cpython-314.pyc b/bird_cnn/__pycache__/server.cpython-314.pyc
index 1046a08..ab31532 100644
Binary files a/bird_cnn/__pycache__/server.cpython-314.pyc and b/bird_cnn/__pycache__/server.cpython-314.pyc differ
diff --git a/bird_cnn/server.py b/bird_cnn/server.py
index 09abdbb..9ebb980 100644
--- a/bird_cnn/server.py
+++ b/bird_cnn/server.py
@@ -38,7 +38,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model = Bird_CNN(c_in=3, c_hidden=15, c_out=7)
 full_path = os.path.join(BUILD_PATH, "bird_cnn")
-model.load_state_dict(torch.load(full_path, weights_only=False))
+model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
 model.to(device)
 model.eval()