Create README.md

README.md

@@ -0,0 +1,37 @@
+---
+license: mit
+tags:
+- medical
+---
+# ViT Brain Tumor Classifier 🧠
+
+A ViT model trained on 75k brain MRI images using PyTorch + TIMM.
+
+## Labels
+- Glioma
+- Meningioma
+- Tumor (general)
+
+## Dataset
+[Brain Cancer MRI Dataset (2024)](https://www.kaggle.com/datasets/shuvokumarbasakbd/brain-cancer-mri-colorized-dataset)
+
+## How to Use
+
+```python
+from PIL import Image
+from torchvision import transforms
+import torch
+import timm
+
+model = timm.create_model('vit_base_patch16_224', pretrained=False, num_classes=3)
+model.load_state_dict(torch.load("pytorch_model.bin"))
+model.eval()
+
+img = Image.open("your_image.jpg")
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.5]*3, std=[0.5]*3),
+])
+input_tensor = transform(img).unsqueeze(0)
+pred = model(input_tensor).argmax(dim=1)