Update README.md

README.md

@@ -1,22 +1,23 @@
 ---
 license: other
 tags:
-  - vision
-  - image-segmentation
+- vision
+- image-segmentation
 datasets:
-  - oxford_pets
+- oxford_pets
 widget:
-  - src: >-
-      https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000001.jpg
-    example_title: Dog
-  - src: >-
-      https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000002.jpg
-    example_title: Cat
+- src: >-
+    https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000001.jpg
+  example_title: Dog
+- src: >-
+    https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000002.jpg
+  example_title: Cat
+library_name: keras
 ---
 
 # sayakpaul/mit-b0-finetuned-pets
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the [Oxford Pets](https://www.robots.ox.ac.uk/~vgg/data/pets/) dataset.
 It achieves the following results on the evaluation set:
 - Train Loss: 0.1481
 - Validation Loss: 0.1962
@@ -62,4 +63,4 @@ The following hyperparameters were used during training:
 
 - Transformers 4.25.1
 - TensorFlow 2.10.1
-- Tokenizers 0.13.2
+- Tokenizers 0.13.2