Update README.md

README.md

@@ -39,19 +39,24 @@ pip install ultralytics
 
 ### How to use
 
-You can do the predictions with the following code snippet:
+You can detect handwritten text with the following code snippet:
 ```
 import ultralytics
+from ultralytics import YOLO
 from huggingface_hub import hf_hub_download
 
-# Loading the weights from our repository
+# Load the weights from our repository
 model_path = hf_hub_download(local_dir=".",
                              repo_id="armvectores/yolov5_handwritten_text_detection",
                              filename="best.pt")
 model = YOLO(model_path)
 
+# Load test blank
+test_blank_path = hf_hub_download(local_dir=".",
+                             repo_id="armvectores/yolov5_handwritten_text_detection",
+                             filename="test_blank.png")
 # Do the predictions
-model.predict(source="path_to_your_image", save=True, show=True, show_labels=False, show_conf=False, conf=0.3)
+model.predict(source=test_blank, save=True, show=True, show_labels=False, show_conf=False, conf=0.3)
 ```