Update README.md

README.md

@@ -1,20 +1,18 @@
 ---
+base_model: FastVIT
 tags:
 - fastai
+- image-classification
+- breast cancer
+metrics:
+- accuracy
 ---
 
-# Amazing!
 
-🥳 Congratulations on hosting your fastai model on the Hugging Face Hub!
-
-# Some next steps
-1. Fill out this model card with more information (see the template below and the [documentation here](https://huggingface.co/docs/hub/model-repos))!
-
-2. Create a demo in Gradio or Streamlit using 🤗 Spaces ([documentation here](https://huggingface.co/docs/hub/spaces)).
-
-3. Join the fastai community on the [Fastai Discord](https://discord.com/invite/YKrxeNn)!
-
-Greetings fellow fastlearner 🤝! Don't forget to delete this content from your model card.
+This model is a fine-tuned version of FastViT on the Mammogram V1 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.1645
+- Accuracy: 0.9517
 
 
 ---
@@ -28,5 +26,24 @@ More information needed
 ## Intended uses & limitations
 More information needed
 
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-04
+- train_batch_size: 32
+- eval_batch_size: 32
+
 ## Training and evaluation data
-More information needed
+| Training Loss | Epoch | Validation Loss | Accuracy |
+|:-------------:|:-----:|:---------------:|:--------:|
+| 0.6094        | 1.0   | 0.6453          | 0.6135   | 
+| 0.4967        | 2.0   | 0.4347          | 0.8259   |
+| 0.3796        | 3.0   | 0.3768          | 0.7799   |
+| 0.2927        | 4.0   | 0.3472          | 0.8463   |
+| 0.2822        | 5.0   | 0.1645          | 0.9517   |
+
+
+## Framework Versions
+- Pytorch 2.3.1+cu121
+- Datasets 2.21.0
+- FastAi