dkagramanyan
commited on
Commit
•
5095bc8
1
Parent(s):
d30b1c7
Update README.md
Browse files
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/
|
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/
|
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)
|