Update README.md

README.md

@@ -9,7 +9,7 @@ widget:
   example_title: Children
 ---
 
-# Yolos-small-crowd
+# Yolos-small-person
 
 YOLOS model fine-tuned on COCO 2017 object detection (118k annotated images). It was introduced in the paper [You Only Look at One Sequence: Rethinking Transformer in Vision through Object Detection](https://arxiv.org/abs/2106.00666) by Fang et al. and first released in [this repository](https://github.com/hustvl/YOLOS). 
 ![model_image](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/yolos_architecture.png)
@@ -44,8 +44,8 @@ import requests
 
 url = "https://latestbollyholly.com/wp-content/uploads/2024/02/Jacob-Gooch.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-image_processor = AutoImageProcessor.from_pretrained("AdamCodd/yolos-small-crowd")
-model = AutoModelForObjectDetection.from_pretrained("AdamCodd/yolos-small-crowd")
+image_processor = AutoImageProcessor.from_pretrained("AdamCodd/yolos-small-person")
+model = AutoModelForObjectDetection.from_pretrained("AdamCodd/yolos-small-person")
 inputs = image_processor(images=image, return_tensors="pt")
 outputs = model(**inputs)