dkagramanyan commited on
Commit
5095bc8
1 Parent(s): d30b1c7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,13 +33,13 @@ from huggingface_hub import hf_hub_download
33
 
34
  # Load the weights from our repository
35
  model_path = hf_hub_download(local_dir=".",
36
- repo_id="armvectores/yolov5_handwritten_text_detection",
37
  filename="best.pt")
38
  model = YOLO(model_path)
39
 
40
  # Load test blank
41
  test_blank_path = hf_hub_download(local_dir=".",
42
- repo_id="armvectores/yolov5_handwritten_text_detection",
43
  filename="test_blank.png")
44
  # Do the predictions
45
  model.predict(source=test_blank, save=True, show=True, show_labels=False, show_conf=False, conf=0.3)
 
33
 
34
  # Load the weights from our repository
35
  model_path = hf_hub_download(local_dir=".",
36
+ repo_id="armvectores/yolov8n_handwritten_text_detection",
37
  filename="best.pt")
38
  model = YOLO(model_path)
39
 
40
  # Load test blank
41
  test_blank_path = hf_hub_download(local_dir=".",
42
+ repo_id="armvectores/yolov8n_handwritten_text_detection",
43
  filename="test_blank.png")
44
  # Do the predictions
45
  model.predict(source=test_blank, save=True, show=True, show_labels=False, show_conf=False, conf=0.3)