Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ from hezar.models import Model
|
|
23 |
from hezar.utils import draw_boxes
|
24 |
|
25 |
|
26 |
-
model = Model.load("
|
27 |
outputs = model.predict("../assets/sample.png")
|
28 |
image = cv2.imread("../assets/O1.png")
|
29 |
result_image = draw_boxes(image, outputs[0]["boxes"])
|
|
|
23 |
from hezar.utils import draw_boxes
|
24 |
|
25 |
|
26 |
+
model = Model.load("hezarai/CRAFT", device="cuda")
|
27 |
outputs = model.predict("../assets/sample.png")
|
28 |
image = cv2.imread("../assets/O1.png")
|
29 |
result_image = draw_boxes(image, outputs[0]["boxes"])
|