Update tasks/image.py

tasks/image.py

@@ -37,7 +37,7 @@ print("CUDA version:", torch.version.cuda)
 print("cuDNN version:", torch.backends.cudnn.version())
 
 #MODEL_TYPE = "YOLOv11n"
-DESCRIPTION = f"YOLOv8n model with batch 64 inference on TensorRT"
+DESCRIPTION = f"YOLOv8n quantisized model with batch 64 inference on TensorRT"
 print(14)
 ROUTE = "/image"
 print(15)
@@ -141,7 +141,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280.onnx'
     #PATH_TO_MODEL = 'models/best_yolov6n_1280.pt'
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
-    PATH_TO_MODEL = 'models/best_YOLOv8n_full_batch_64.engine'
+    PATH_TO_MODEL = 'models/best_YOLOv8n_half_batch_64.engine'
     INFERENCE_ENGINE_TYPE = 'pt'
     INPUT_SIZE = 640
     N_TEST_BATCHES = 2