Agmiyas commited on
Commit
13414e8
1 Parent(s): 0b36a18

README minor changes

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -42,13 +42,13 @@ pip install ultralytics
42
  You can do the predictions with the following code snippet:
43
  ```
44
  import ultralytics
45
- import detector
46
  from huggingface_hub import hf_hub_download
47
 
 
48
  model_path = hf_hub_download(local_dir=".",
49
  repo_id="armvectores/yolov5_handwritten_text_detection",
50
  filename="best.pt")
51
- model = detector.load_model(model_path)
52
 
53
  # Do the predictions
54
  model.predict(source="path_to_your_image", save=True, show=True, show_labels=False, show_conf=False, conf=0.3)
 
42
  You can do the predictions with the following code snippet:
43
  ```
44
  import ultralytics
 
45
  from huggingface_hub import hf_hub_download
46
 
47
+ # Loading the weights from our repository
48
  model_path = hf_hub_download(local_dir=".",
49
  repo_id="armvectores/yolov5_handwritten_text_detection",
50
  filename="best.pt")
51
+ model = YOLO(model_path)
52
 
53
  # Do the predictions
54
  model.predict(source="path_to_your_image", save=True, show=True, show_labels=False, show_conf=False, conf=0.3)