Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- 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(
|
|
|
|
|
|
|
| 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 |
|