Frugal_AI_image_2025

Sleeping

tom-b974 commited on Jan 27

Commit

8895a62

verified ·

1 Parent(s): 3ae7280

Update tasks/image.py

Files changed (1) hide show

tasks/image.py CHANGED Viewed

@@ -18,7 +18,7 @@ router = APIRouter()
 DESCRIPTION = "YOLO Smoke Detection"
 ROUTE = "/image"
-yolo_model = YOLO("yolo11_tr_20012025_frugalAIchal.pt")
 def parse_boxes(annotation_string):
     """Parse multiple boxes from a single annotation string.

 DESCRIPTION = "YOLO Smoke Detection"
 ROUTE = "/image"
+yolo_model = YOLO("yolo11_vl_20012025_frugalAIchal.pt")
 def parse_boxes(annotation_string):
     """Parse multiple boxes from a single annotation string.