Update README.md

README.md

@@ -10,6 +10,7 @@ library_name: transformers
 tags:
 - object detection
 - vision
+- yolo
 pipeline_tag: object-detection
 ---
 
@@ -30,11 +31,15 @@ You can use the raw model for object detection on cashew images.
 
 - Load model and perform prediction:
 
+```bash
+pip install -U ultralytics
+```
+
 ```python
 import torch
 
 # load model
-model = torch.hub.load('ultralytics/yolov5', 'KaraAgroAI/CADI-AI')
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='CADI-AI/yolov5_0.65map_exp7_best.pt',  force_reload=True)
 
 # Images
 img = ['/path/to/CADI-AI-image.jpg']# batch of images