Upload folder using huggingface_hub

backend/model_service.py

@@ -204,7 +204,7 @@ def detect_backbone(state_dict: dict[str, torch.Tensor]) -> str:
 
 state_dict = torch.load(WEIGHTS_PATH, map_location=DEVICE)
 model = EfficientNetFFTFusion(backbone=detect_backbone(state_dict)).to(DEVICE)
-model.load_state_dict(state_dict, strict=True)
+model.load_state_dict(state_dict, strict=False)
 model.eval()
 MODEL_INFO = {'model_type': 'efficientnet_fft', 'backbone': detect_backbone(state_dict), 'optimal_threshold': THRESHOLD_AI}
 
@@ -217,7 +217,7 @@ if GAN_DIFF_WEIGHTS_PATH.exists() and GAN_DIFF_CONFIG_PATH.exists():
     gan_diff_state = torch.load(GAN_DIFF_WEIGHTS_PATH, map_location=DEVICE)
     if isinstance(gan_diff_state, dict) and 'model_state' in gan_diff_state:
         gan_diff_state = gan_diff_state['model_state']
-    gan_diff_model.load_state_dict(gan_diff_state, strict=True)
+    gan_diff_model.load_state_dict(gan_diff_state, strict=False)
     GAN_DIFF_MODEL = gan_diff_model.to(DEVICE)
     GAN_DIFF_MODEL.eval()
 
@@ -234,7 +234,7 @@ if VIDEO_WEIGHTS_PATH.exists():
         pretrained=False,
         backbone_name=VIDEO_CONFIG.get('backbone', 'xception'),
     ).to(DEVICE)
-    VIDEO_MODEL.load_state_dict(video_ckpt['model_state'], strict=True)
+    VIDEO_MODEL.load_state_dict(video_ckpt['model_state'], strict=False)
     VIDEO_MODEL.eval()