Update README.md

README.md

@@ -16,7 +16,7 @@ YOLOS model fine-tuned on COCO 2017 object detection (118k annotated images). It
 
 
 ## Model description
-This model has been finetuned on the following [crowd-detection dataset](https://universe.roboflow.com/institut-teknologi-nasional-bandung-mxgtc/crowd-detection-i75bl) with the following results on the test set:
+This model has been finetuned on the following [crowd-detection dataset](https://universe.roboflow.com/institut-teknologi-nasional-bandung-mxgtc/crowd-detection-i75bl) (669 samples) with the following results on the test set:
 
 ```
 IoU metric: bbox
@@ -64,7 +64,7 @@ Refer to the [documentation](https://huggingface.co/docs/transformers/main/en/mo
 
 ## Intended uses & limitations
 
-This fine-tuned model performs best when detecting individuals who are relatively close (but not too close) to the viewpoint. As indicated by the metrics, it struggles to identify individuals farther away.
+This model is more of an experiment on a small scale and will need retraining on a more diverse dataset. This fine-tuned model performs best when detecting individuals who are relatively close (but not too close) to the viewpoint. As indicated by the metrics, it struggles to identify individuals farther away.
 
 ### Training hyperparameters