Update README.md

README.md

@@ -30,6 +30,7 @@ pip install ultralytics
 ```
 from ultralytics import YOLO
 from huggingface_hub import hf_hub_download
+from matplotlib import pyplot as plt
 
 # Load the weights from our repository
 model_path = hf_hub_download(local_dir=".",
@@ -41,8 +42,13 @@ model = YOLO(model_path)
 test_blank_path = hf_hub_download(local_dir=".",
                              repo_id="armvectores/yolov8n_handwritten_text_detection",
                              filename="test_blank.png")
+
 # Do the predictions
-model.predict(source=test_blank, save=True, show=True, show_labels=False, show_conf=False, conf=0.3)
+res = model.predict(source=test_blank_path, project='.',name='detected', exist_ok=True, save=True, show=False, show_labels=False, show_conf=False, conf=0.5, )
+
+plt.figure(figsize=(15,10))
+plt.imshow(plt.imread('detected/test_blank.png'))
+plt.show()
 ```