Update README.md

README.md

@@ -1,28 +1,36 @@
-
-# Model Card for Cats vs Dogs Classifier
-
-## Model Description
-This model is a Convolutional Neural Network (CNN) designed to classify images of cats and dogs. It was trained on the "louiecerv/cats_dogs_dataset" dataset.
-
-## Intended Use
-The model is intended for educational purposes and as a demonstration of image classification using CNNs. It can be used to classify images of cats and dogs.
-
-## Training Data
-The model was trained on the "louiecerv/cats_dogs_dataset" dataset, which contains images of cats and dogs.
-
-## Training Procedure
-The model was trained using the following procedure:
-- Optimizer: Adam
-- Learning Rate: 0.001
-- Loss Function: CrossEntropyLoss
-- Number of Epochs: 10
-- Batch Size: 32
-
-## Evaluation
-The model was evaluated on the training dataset. Further evaluation on a separate validation or test dataset is recommended.
-
-## Limitations
-The model may not perform well on images that are significantly different from the training data. It is also limited to binary classification (cats vs. dogs).
-
-## License
-The model is licensed under the MIT License.
+---
+license: mit
+datasets:
+- louiecerv/cats_dogs_dataset
+language:
+- en
+pipeline_tag: image-classification
+---
+
+# Model Card for Cats vs Dogs Classifier
+
+## Model Description
+This model is a Convolutional Neural Network (CNN) designed to classify images of cats and dogs. It was trained on the "louiecerv/cats_dogs_dataset" dataset.
+
+## Intended Use
+The model is intended for educational purposes and as a demonstration of image classification using CNNs. It can be used to classify images of cats and dogs.
+
+## Training Data
+The model was trained on the "louiecerv/cats_dogs_dataset" dataset, which contains images of cats and dogs.
+
+## Training Procedure
+The model was trained using the following procedure:
+- Optimizer: Adam
+- Learning Rate: 0.001
+- Loss Function: CrossEntropyLoss
+- Number of Epochs: 10
+- Batch Size: 32
+
+## Evaluation
+The model was evaluated on the training dataset. Further evaluation on a separate validation or test dataset is recommended.
+
+## Limitations
+The model may not perform well on images that are significantly different from the training data. It is also limited to binary classification (cats vs. dogs).
+
+## License
+The model is licensed under the MIT License.