use YOLOv11n

models/best_yolov11n.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:729c35de693cee7ddf5f86fe350b6937d3cd7013fb483ea6636ab7a11c52ad32
+size 38121666

tasks/image.py

@@ -15,7 +15,8 @@ load_dotenv()
 
 router = APIRouter()
 
-DESCRIPTION = "Simple YOLOv8n model"
+MODEL_TYPE = "YOLOv11n"
+DESCRIPTION = f"Simple {MODEL_TYPE} model"
 ROUTE = "/image"
 
 def parse_boxes(annotation_string):
@@ -112,7 +113,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     
     
-    PATH_TO_MODEL = "models/best_YOLOv8n.pt"
+    PATH_TO_MODEL = f"models/best_{MODEL_TYPE}.pt"
     model = load_model(PATH_TO_MODEL)
     
     print(f"Model info: {model.info()}")