levydaniel commited on
Commit
b0aea59
·
verified ·
1 Parent(s): 15b851e

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. tasks/image.py +1 -1
tasks/image.py CHANGED
@@ -17,7 +17,7 @@ router = APIRouter()
17
  DESCRIPTION = "Random Baseline"
18
  ROUTE = "/image"
19
 
20
- model = YOLO("test_best.pt")
21
 
22
  def parse_boxes(annotation_string):
23
  """Parse multiple boxes from a single annotation string.
 
17
  DESCRIPTION = "Random Baseline"
18
  ROUTE = "/image"
19
 
20
+ model = YOLO("best_test.pt")
21
 
22
  def parse_boxes(annotation_string):
23
  """Parse multiple boxes from a single annotation string.