update model card README.md

README.md

@@ -1,10 +1,8 @@
 ---
 license: apache-2.0
-base_model: facebook/detr-resnet-50
+base_model: akar49/detr-resnet-50_machinery-I
 tags:
 - generated_from_trainer
-datasets:
-- excavator-detector
 model-index:
 - name: detr-resnet-50_machinery-I
   results: []
@@ -15,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # detr-resnet-50_machinery-I
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the excavator-detector dataset.
+This model is a fine-tuned version of [akar49/detr-resnet-50_machinery-I](https://huggingface.co/akar49/detr-resnet-50_machinery-I) on an unknown dataset.
 
 ## Model description
 
@@ -49,6 +47,6 @@ The following hyperparameters were used during training:
 ### Framework versions
 
 - Transformers 4.31.0
-- Pytorch 2.0.1+cu118
-- Datasets 2.14.4
+- Pytorch 2.1.0+cu121
+- Datasets 2.16.0
 - Tokenizers 0.13.3