README minor changes
Browse files
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 =
|
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)
|