CaffeinatedCoding commited on
Commit
4756353
·
verified ·
1 Parent(s): 7df0705

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. src/xai.py +3 -1
src/xai.py CHANGED
@@ -65,7 +65,9 @@ class GradCAMPlusPlus:
65
  )
66
  else:
67
  # Fallback: pretrained ImageNet weights (weaker XAI but not None)
68
- self.model = models.efficientnet_b0(pretrained=True)
 
 
69
  print("WARNING: EfficientNet fine-tuned weights not found. "
70
  "Using ImageNet pretrained — GradCAM++ quality reduced.")
71
 
 
65
  )
66
  else:
67
  # Fallback: pretrained ImageNet weights (weaker XAI but not None)
68
+ self.model = models.efficientnet_b0(
69
+ weights=models.EfficientNet_B0_Weights.IMAGENET1K_V1
70
+ )
71
  print("WARNING: EfficientNet fine-tuned weights not found. "
72
  "Using ImageNet pretrained — GradCAM++ quality reduced.")
73