Upload inference.py

inference.py

@@ -113,7 +113,9 @@ class DeepFauneModel(nn.Module):
             raise FileNotFoundError(f"Model file not found: {self.model_path}")
 
         try:
-            params = torch.load(self.model_path, map_location=device)
+            # torch 2.6+ flips weights_only to True; this trusted checkpoint
+            # pickles a torchvision Compose, so keep the old full load.
+            params = torch.load(self.model_path, map_location=device, weights_only=False)
             args = params['args']
 
             # Validate number of classes matches