Upload model.py with huggingface_hub

model.py

@@ -63,10 +63,12 @@ def load_classification_model(provider: str = "google", device: str = "cpu"):
     assert provider in ("google", "ign"), "provider must be 'google' or 'ign'"
 
     path  = hf_hub_download(REPO, f"inception_{provider}_best.pth")
-    model = inception_v3(weights=None, aux_logits=False)
-    model.fc = nn.Linear(model.fc.in_features, 1)
+    model = inception_v3(weights=None, aux_logits=True)
+    model.fc           = nn.Linear(model.fc.in_features, 1)
+    model.AuxLogits.fc = nn.Linear(model.AuxLogits.fc.in_features, 1)
 
     state = torch.load(path, map_location=device, weights_only=False)
     model.load_state_dict(state)
+    model.aux_logits = False  # désactive pour l'inférence
 
     return model.eval().to(device)