README changes

README.md

@@ -28,10 +28,13 @@ pip install ultralytics
 You can do the predictions with the following code snippet:
 ```
 import ultralytics
-from ultralytics import YOLO
+import detector
+from huggingface_hub import hf_hub_download
 
-# Load weights for YOLO from our repository
-model = YOLO('best.pt')
+model_path = hf_hub_download(local_dir=".",
+                             repo_id="armvectores/yolov5_handwritten_text_detection",
+                             filename="best.pt")
+model = detector.load_model(model_path)
 
 # Do the predictions
 model.predict(source="path_to_your_image", save=True, show=True, show_labels=False, show_conf=False, conf=0.3)