arxyzan commited on
Commit
8aef362
1 Parent(s): 2e9091e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("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"])
 
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"])