Update README.md

README.md

@@ -1,6 +1,24 @@
 ---
 library_name: transformers
+license: apache-2.0
+language:
+  - en
 pipeline_tag: object-detection
+tags:
+  - object-detection
+  - vision
+datasets:
+  - coco
+widget:
+  - src: >-
+      https://huggingface.co/datasets/mishig/sample_images/resolve/main/savanna.jpg
+    example_title: Savanna
+  - src: >-
+      https://huggingface.co/datasets/mishig/sample_images/resolve/main/football-match.jpg
+    example_title: Football Match
+  - src: >-
+      https://huggingface.co/datasets/mishig/sample_images/resolve/main/airport.jpg
+    example_title: Airport
 ---
 ## RT-DETRv2
 
@@ -30,8 +48,8 @@ from transformers import RTDetrV2ForObjectDetection, RTDetrImageProcessor
 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
 
-image_processor = RTDetrImageProcessor.from_pretrained("jadechoghari/rtdetr_v2_r18vd")
-model = RTDetrV2ForObjectDetection.from_pretrained("jadechoghari/rtdetr_v2_r18vd")
+image_processor = RTDetrImageProcessor.from_pretrained("PekingU/rtdetr_v2_r18vd")
+model = RTDetrV2ForObjectDetection.from_pretrained("PekingU/rtdetr_v2_r18vd")
 
 inputs = image_processor(images=image, return_tensors="pt")