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()}")
|