Spaces:
Sleeping
Sleeping
fix(): remove dynamic description
Browse files- tasks/image.py +3 -3
tasks/image.py
CHANGED
@@ -15,8 +15,8 @@ load_dotenv()
|
|
15 |
|
16 |
router = APIRouter()
|
17 |
|
18 |
-
MODEL_TYPE = "YOLOv11n"
|
19 |
-
DESCRIPTION = f"Simple
|
20 |
ROUTE = "/image"
|
21 |
|
22 |
def parse_boxes(annotation_string):
|
@@ -113,7 +113,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
|
|
113 |
#--------------------------------------------------------------------------------------------
|
114 |
|
115 |
|
116 |
-
PATH_TO_MODEL = f"models/
|
117 |
model = load_model(PATH_TO_MODEL)
|
118 |
|
119 |
print(f"Model info: {model.info()}")
|
|
|
15 |
|
16 |
router = APIRouter()
|
17 |
|
18 |
+
#MODEL_TYPE = "YOLOv11n"
|
19 |
+
DESCRIPTION = f"Simple YOLOv11n model"
|
20 |
ROUTE = "/image"
|
21 |
|
22 |
def parse_boxes(annotation_string):
|
|
|
113 |
#--------------------------------------------------------------------------------------------
|
114 |
|
115 |
|
116 |
+
PATH_TO_MODEL = f"models/best_YOLOv11n.pt"
|
117 |
model = load_model(PATH_TO_MODEL)
|
118 |
|
119 |
print(f"Model info: {model.info()}")
|