Update README.md

README.md

@@ -17,7 +17,7 @@ datasets:
 - OpenNeuro
 base_model:
 - google/medgemma-1.5-4b-it
-- google/medsiglip-base-patch16-448
+- google/medsiglip-448
 ---
 
 # 🧠 BrainGemma3D — Brain Report Automation via Inflated Vision Transformers in 3D
@@ -76,7 +76,6 @@ pip install torch torchvision transformers nibabel scikit-image lime
 ```python
 from huggingface_hub import snapshot_download
 
-
 # 1. Download the repository containing our custom architecture from Hugging Face
 repo_id = "praiselab-picuslab/BrainGemma3D"
 print(f"Downloading repository: {repo_id}...")
@@ -98,7 +97,6 @@ from medgemma3d_architecture import MedGemma3D, load_nifti_volume, CANONICAL_PRO
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Hardware accelerator selected: {device}")
 
-
 # 2. Instantiate the base architecture (3D-inflated MedSigLIP + MedGemma)
 model = MedGemma3D(
     vision_model_dir=f"{local_dir}/vision_model",
@@ -110,7 +108,6 @@ model = MedGemma3D(
     device_map={"": 0} if device == "cuda" else None,
 )
 
-
 # 3. Load projector
 proj_path = os.path.join(local_dir, "projector_vis_scale.pt")
 print(f"Loading custom projector weights from: {proj_path}...")
@@ -132,7 +129,6 @@ if ckpt.get("vis_scale") is not None:
 model.eval()
 print("✅ BrainGemma3D is fully loaded and ready for inference!")
 
-
 # 4. Load MRI scan
 volume = load_nifti_volume(
     "path/to/brain_flair.nii.gz",
@@ -256,6 +252,13 @@ weights, wvol = run_interpretability(
 - `lime_top_supervoxels_grid.png` — Most influential supervoxels
 - `lime_weights.json` — Supervoxel importance scores
 
+### Expected Output Example
+
+<div align="left">
+    <img src="https://cdn-uploads.huggingface.co/production/uploads/662a12d70951c58269b066fb/UkQwmZRwkn-rlNlFBNVkH.png" alt="LIME Interpretability" width="50%">
+    <p><i>Figure 3: LIME attribution maps for BraTS training sample 072. Red regions show supervoxels that positively contribute to pathology predictions. The model correctly focuses on tumor-affected areas in the left parietal and frontal lobes.</i></p>
+</div>
+
 ---
 
 ## ⚙️ Model Details