dhineshkmar commited on
Commit
f6368f7
·
verified ·
1 Parent(s): b001502

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +40 -0
README.md CHANGED
@@ -1,3 +1,43 @@
1
  ---
2
  license: mit
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+ - medical
5
+ - brain-disease
6
+ - image-classification
7
+ - multiplesclerosis
8
  ---
9
+
10
+ # Multiplesclerosis Detection Model
11
+
12
+ This model detects multiplesclerosis from brain MRI scans.
13
+
14
+ ## Model Details
15
+ - **Architecture**: VIT
16
+ - **Task**: Image Classification
17
+ - **Classes**: 4
18
+ - **Input Size**: 224x224
19
+
20
+ ## Labels
21
+ - 0: Control-Axial
22
+ - 1: Control-Sagittal
23
+ - 2: MS-Axial
24
+ - 3: MS-Sagittal
25
+
26
+ ## Usage
27
+
28
+ ```python
29
+ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
30
+ from PIL import Image
31
+
32
+ model_name = "dhineshkmar/multiplesclerosis"
33
+ feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
34
+ model = AutoModelForImageClassification.from_pretrained(model_name)
35
+
36
+ image = Image.open("brain_scan.jpg")
37
+ inputs = feature_extractor(images=image, return_tensors="pt")
38
+ outputs = model(**inputs)
39
+ predicted_class = outputs.logits.argmax(-1).item()
40
+ ```
41
+
42
+ ## Training
43
+ Trained on brain MRI scan dataset for multiplesclerosis detection.