update model card README.md

README.md

@@ -1,6 +1,7 @@
 ---
 license: apache-2.0
 tags:
+- image-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -14,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Train
 
-This model is a fine-tuned version of [microsoft/resnet-101](https://huggingface.co/microsoft/resnet-101) on the None dataset.
+This model is a fine-tuned version of [microsoft/resnet-101](https://huggingface.co/microsoft/resnet-101) on the TrashBox dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.8217
-- Accuracy: 0.3872
+- Loss: 2.7726
+- Accuracy: 0.4374
 
 ## Model description