Update app.py

app.py

@@ -143,7 +143,7 @@ def predict_with_tta(image, n_augmentations=3):
     return mean_pred, std_pred
 
 def generate_gradcam(image, disease_idx):
-    """Generate Grad-CAM visualization for specific disease"""
+    """Generate improved Grad-CAM visualization for specific disease"""
     if isinstance(image, np.ndarray):
         img_pil = Image.fromarray(image.astype('uint8'))
     else:
@@ -153,11 +153,11 @@ def generate_gradcam(image, disease_idx):
     img_array = np.array(img_resized) / 255.0
     img_array = np.expand_dims(img_array, axis=0).astype(np.float32)
     
-    # Generate heatmap
+    # Generate improved heatmap with noise reduction
     heatmap = make_gradcam_heatmap(img_array, model, last_conv_layer, disease_idx)
     
-    # Overlay on original image
-    overlaid_image = overlay_heatmap_on_image(img_resized, heatmap, alpha=0.4)
+    # Overlay with better alpha for medical images
+    overlaid_image = overlay_heatmap_on_image(img_resized, heatmap, alpha=0.5)
     
     return overlaid_image