README.md

metadata

title: Pneumonia Detector
emoji: 🫁
colorFrom: purple
colorTo: red
sdk: gradio
sdk_version: 6.12.0
app_file: app.py
pinned: false
license: mit
short_description: Pneumonia detection from chest X-rays using ViT and Grad-CAM

🫁 Pneumonia Detector

Medical AI system for detecting pneumonia from chest X-ray images using Vision Transformer (ViT) with explainable Grad-CAM heatmaps.

Features

• Upload chest X-ray — supports PNG, JPEG
• AI prediction — Normal or Pneumonia with confidence score
• Grad-CAM heatmap — visual explanation showing which regions the model focused on
• Fast inference — runs on CPU or GPU

Model Details

Attribute	Value
Architecture	Vision Transformer (ViT)
Backbone	ViT-B/16
Fine-tuned on	Chest X-ray dataset
Accuracy	87%
AUC-ROC	0.98
Input size	224x224 pixels

Training Details

Parameter	Value
Dataset	Chest X-ray (Pneumonia)
Train/Val/Test split	80/10/10
Epochs	20
Batch size	32
Learning rate	1e-4
Optimizer	AdamW
Hardware	Kaggle P100 (16GB)
Training time	~3 hours

Performance Metrics

Metric	Score
Accuracy	87%
AUC-ROC	0.98
Precision	85%
Recall	89%
F1-Score	87%

Tested on held-out test set of 624 images

How to Use

1. Upload a chest X-ray image (frontal view preferred)
2. Click Submit
3. View prediction + Grad-CAM heatmap

Example Output

Prediction: Pneumonia (confidence: 94%)
Heatmap: Highlights upper right lung region

Technical Stack

• PyTorch
• Hugging Face Transformers
• timm (Vision Models)
• Grad-CAM
• Gradio

Run Locally

git clone https://huggingface.co/spaces/mou11/pneumonia-detector
cd pneumonia-detector
pip install -r requirements.txt
python app.py