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)
|