Update shadow_generator.py

shadow_generator.py

@@ -341,10 +341,17 @@ class SSNWrapper:
         _validate_assets()
         ssn_model_dir = _find_ssn_model_file()
         sys.path.insert(0, str(ssn_model_dir.resolve()))
+
         try:
             from SSN_Model import SSN_Model
             self.model = SSN_Model()
-            state = torch.load(str(WEIGHT_FILE), map_location=self.device)
+
+            # ✅ Patch for numpy scalar safety
+            torch.serialization.add_safe_globals([np.core.multiarray.scalar])
+
+            # ✅ Allow full checkpoint load
+            state = torch.load(str(WEIGHT_FILE), map_location=self.device, weights_only=False)
+
             if isinstance(state, dict):
                 if "model_state_dict" in state:
                     sd = state["model_state_dict"]
@@ -356,6 +363,7 @@ class SSNWrapper:
                     sd = state
             else:
                 sd = state
+
             self.model.load_state_dict(sd, strict=False)
             self.model.eval().to(self.device)
             _log(f"✅ SSN model loaded on {self.device}")