Update README.md

README.md

@@ -12,6 +12,9 @@ metrics:
 model-index:
 - name: levit_128.fb_dist_in1k-finetuned-stroke-binary
   results: []
+datasets:
+- BTX24/tekno21-brain-stroke-dataset-binary
+pipeline_tag: image-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -19,26 +22,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # levit_128.fb_dist_in1k-finetuned-stroke-binary
 
-This model is a fine-tuned version of [timm/levit_128.fb_dist_in1k](https://huggingface.co/timm/levit_128.fb_dist_in1k) on an unknown dataset.
+This model is a fine-tuned version of [timm/levit_128.fb_dist_in1k](https://huggingface.co/timm/levit_128.fb_dist_in1k) on an binary stroke detection dataset.
 It achieves the following results on the evaluation set:
-- Loss: nan
 - Accuracy: 0.8598
 - F1: 0.8577
 - Precision: 0.8602
 - Recall: 0.8598
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
 ## Training procedure
 
 ### Training hyperparameters
@@ -80,4 +70,4 @@ The following hyperparameters were used during training:
 - Transformers 4.48.3
 - Pytorch 2.6.0+cu124
 - Datasets 3.4.0
-- Tokenizers 0.21.0
+- Tokenizers 0.21.0