Upload README.md with huggingface_hub

README.md

@@ -1,3 +1,41 @@
 ---
 license: mit
+tags:
+- medical
+- brain-disease
+- image-classification
+- stroke
 ---
+
+# Stroke Detection Model
+
+This model detects stroke from brain MRI scans.
+
+## Model Details
+- **Architecture**: CONVNEXT
+- **Task**: Image Classification
+- **Classes**: 2
+- **Input Size**: 224x224
+
+## Labels
+- 0: Normal
+- 1: Stroke
+
+## Usage
+
+```python
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from PIL import Image
+
+model_name = "dhineshkmar/stroke"
+feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
+model = AutoModelForImageClassification.from_pretrained(model_name)
+
+image = Image.open("brain_scan.jpg")
+inputs = feature_extractor(images=image, return_tensors="pt")
+outputs = model(**inputs)
+predicted_class = outputs.logits.argmax(-1).item()
+```
+
+## Training
+Trained on brain MRI scan dataset for stroke detection.