weird fixes

src/draft_model.py

@@ -20,7 +20,7 @@ ENCODING_FILE = "card_encodings.pt"
 
 
 class DraftModel:
-    def __init__(self, model_path: str):
+    def __init__(self):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
         cfg_path = hf_hub_download(
@@ -35,7 +35,7 @@ class DraftModel:
         cfg.pop("name", None)
 
         self.net = MLP_CrossAttention(**cfg).to(self.device)
-        self.net.load_state_dict(torch.load(Path(model_path) / "network.pt", map_location=self.device))
+        self.net.load_state_dict(weight_path, map_location=self.device))
         self.net.eval()
 
         # ---- embeddings – one-time load ------------------------------------