Update README.md

README.md

@@ -1,3 +1,58 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+datasets:
+- Arunisto/brain_tumor_dataset
+language:
+- en
+base_model:
+- microsoft/swin-tiny-patch4-window7-224
+pipeline_tag: image-classification
+library_name: transformers
+tags:
+- computer_vision
+- image_classification
+---
+
+# Pre-trained swin transformer model for brain tumor classification
+
+## Model Description
+This is pre-trained swintransformer model for classify brain tumor.
+
+## Training Dataset
+This model trained with MRI Scan images of brain  tumor `healthy` and `tumor`
+
+### Dataset Format
+- **Images:** JPEG/PNG formats
+- **Dataset:** Arunisto/brain_tumor_dataset
+
+## Training Details
+- **Model:** microsoft/swin-tiny-patch4-window7-224
+- **Framework:** Transformers
+- **Epochs:** 10
+
+
+## Usage
+This model can be used to detect two type of brain condition is `healthy` or `tumor`
+
+
+## Limitations
+- The model is perform well only with brain top-side only, and for healthy it's only predicted with the score of 65% only 
+
+## Future Work
+- needs to run more epochs
+- trying to classify different tumor conditions
+  
+## License
+Apache 2.0
+
+## Citation
+If you use this model in your research or projects, please cite it as follows:
+
+```
+@misc{arun-arunisto2024yolov8_bike_odometer,
+  title={Pre-trained swin transformer model for brain tumor classification},
+  author={Arun Arunisto},
+  year={2024},
+  howpublished={\url{https://huggingface.co/Arunisto/brain_tumor_classification}},
+}
+```