Update README.md

README.md

@@ -17,8 +17,9 @@ tags:
 
 ### Model Description
 
-Object detection model trained using YOLO v5x. The model was pre-trained on the Cashew Disease Identification with AI (CADI-AI) train set (3788 images) at a resolution of 640x640 pixels.
-CADI-AI dataset is available in hugging face dataset hub
+Object detection model trained using [YOLO v5x](https://github.com/ultralytics/yolov5/releases), a SOTA object detection algorithm.  
+The model was pre-trained on the Cashew Disease Identification with AI (CADI-AI) train set (3788 images) at a resolution of 640x640 pixels. 
+CADI-AI dataset is available in hugging face dataset hub.
 
 ## Intended uses & limitations
 
@@ -38,6 +39,7 @@ model = torch.hub.load('ultralytics/yolov5', 'KaraAgroAI/CADI-AI')
 img = ['/path/to/CADI-AI-image.jpg']# batch of images
   
 # set model parameters
+# set Non-Maximum-Suppression(NMS) threshold to define minimum confidence score that a bounding box must have in order to be kept.
 model.conf = 0.20  # NMS confidence threshold
 
 # perform inference