From a9cb14cf421cd22dcf666e67aceee0a5104cb296 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marvin=20Krau=C3=9Fer?= <koro.no.omo@gmail.com>
Date: Wed, 29 Apr 2026 14:53:47 +0200
Subject: [PATCH] changed weight loading

---
 bird_cnn/__pycache__/server.cpython-314.pyc | Bin 3610 -> 3650 bytes
 bird_cnn/server.py                          |   2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/bird_cnn/__pycache__/server.cpython-314.pyc b/bird_cnn/__pycache__/server.cpython-314.pyc
index 1046a08213e9562ceced6e05cfbed35a8edfd1b8..ab315322091e1051fc340c53d42a3d4b2a717cec 100644
GIT binary patch
delta 181
zcmbOwb4Z3)n~#@^0SKJ_f6RO`kyny2Xrg)xqr}EFrx+(sV3g#RF<}HrGcW}4Pp)E0
zW)zrwk4c_eFi1#`5sZZ=b1*BKiojIGawswciNbisj6q`J3_;@J3^@Whj3y9+7)rE*
zn3NebB{#=0>oEyvGT!3JO)QAd$xlu!$;{8&JeQS;ow0iI0-oRe`(<{?97wxt>3c=Y
M?+Sze<b}MI0ISj}cK`qY

delta 162
zcmX>kGfRe7n~#@^0SLIhe$2c&kynz@Xrg)xqvXalrx;m+_=5x{&tghu6rB8zNuFCM
zNLY^%j727EFe{phnlOTNGBCt)C^7_z!Fa}uLE_>JK@#E&IRZJ1CJ-eICE7tu$_$!P
zo9meMm;~K48E^5Fr)H*SloZG3=jBvx-owho&R8^gKhJNrD`NIn7#t>_<gEk%J|88~

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()