Update README.md

README.md

@@ -1,23 +1,17 @@
 ---
 tags:
 - fastai
+- image classification
+- breast cancer
+metrics:
+- accuracy
+- f1
 ---
 
-# 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.8279
+- Accuracy: 0.5893
 
 
 # Model card
@@ -28,5 +22,18 @@ More information needed
 ## Intended uses & limitations
 More information needed
 
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 16
+- eval_batch_size: 16
+
 ## Training and evaluation data
-More information needed
+
+
+
+## Framework Versions
+- Pytorch 2.3.1+cu121
+- Datasets 2.21.0
+- FastAi