Add trained model and evaluation artifacts

README.md

@@ -1,11 +1,25 @@
+---
+language:
+- en
+license: apache-2.0
+tags:
+- image-classification
+- pytorch
+- resnet
+- beans
+datasets:
+- beans
+library_name: pytorch
+pipeline_tag: image-classification
+---
 
 # ResNet18 fine-tuned on Beans dataset
 
 This model was trained in Google Colab using a GPU and tracked with MLflow.
 
-Dataset: [Beans](https://huggingface.co/datasets/AI-Lab-Makerere/beans)
+Dataset: [Beans](https://huggingface.co/datasets/beans)
 
-### Classes:
+## Classes:
 - Healthy
 - Bean Rust
 - Angular Leaf Spot