Spaces:

dhruv2842
/

glaucoma_backend_2

Sleeping

dhruv2842 commited on Jun 25

Commit

edcb5de

verified ·

1 Parent(s): 9aea561

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ def predict():
         confidence = float(probabilities[class_index])
         # ✅ Grad-CAM++ setup
-        target_layer = model.features[0].norm5
         cam_model = GradCAMPlusPlus(model=model, target_layers=[target_layer])
         # ✅ Get Grad-CAM++ map

         confidence = float(probabilities[class_index])
         # ✅ Grad-CAM++ setup
+        target_layer = model.features[2].global_spatial_conv
         cam_model = GradCAMPlusPlus(model=model, target_layers=[target_layer])
         # ✅ Get Grad-CAM++ map