Update README.md

README.md

@@ -6,6 +6,8 @@ tags:
 model-index:
 - name: detr-resnet-50_finetuned_detect-waste
   results: []
+datasets:
+- Yorai/detect-waste
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # detr-resnet-50_finetuned_detect-waste
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the None dataset.
+This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the [detect-waste](https://huggingface.co/datasets/Yorai/detect-waste) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.0141
 - Map: 0.2906
@@ -196,4 +198,4 @@ The following hyperparameters were used during training:
 - Transformers 4.33.0.dev0
 - Pytorch 2.0.1+cu117
 - Datasets 2.14.4
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3