Update app.py

app.py

@@ -22,6 +22,24 @@ if uploaded_file:
 
     probs = predict(model, img_tensor, device)
 
+    # Get top class
+    top_disease = max(probs, key=probs.get)
+    target_idx = list(probs.keys()).index(top_disease)
+
+# Grad-CAM
+    cam = generate_gradcam(model, img_tensor, target_idx, device)
+
+# Overlay on image
+    image_resized = image.resize((224, 224))
+    img_np = np.array(image_resized)
+    heatmap = cv2.applyColorMap(np.uint8(255 * cam), cv2.COLORMAP_JET)
+    overlay = cv2.addWeighted(img_np, 0.6, heatmap, 0.4, 0)
+
+# Show it
+    st.subheader(f"Grad-CAM Visualization: {top_disease}")
+    st.image(overlay, use_column_width=True)
+
+
     st.subheader("Predictions")
     for disease, prob in probs.items():
         st.write(f"**{disease}**: {prob:.4f}")