[ref] update to pruned model

app.py

@@ -16,8 +16,8 @@ app.include_router(image.router)
 @app.get("/")
 async def root():
     return {
-        "message": "This is an API for a frugal object detection model",
+        "message": "This is an API for a frugal fire smoke detection model",
         "endpoints": {
-            "image": "/image - Image classification task (coming soon)",
+            "image": "/image - Image classification task",
         }
     } 

tasks/image.py

@@ -103,7 +103,8 @@ async def evaluate_image(request: ImageEvaluationRequest):
 
     # Load model
     model_path = Path("tasks", "models")
-    model_name = "best.pt"
+    # model_name = "best.pt" # nano 20e, unpruned
+    model_name = "pruned.pt" # nano 20e, 20% pruned
     model = YOLO(Path(model_path, model_name))
     threshold = 0.14