Update tasks/image.py

tasks/image.py

@@ -16,7 +16,7 @@ load_dotenv()
 router = APIRouter()
 
 #MODEL_TYPE = "YOLOv11n"
-DESCRIPTION = f"YOLOv8n model with batch 128 inference"
+DESCRIPTION = f"YOLOv8n model with batch 8 inference on CPU"
 
 ROUTE = "/image"
 
@@ -122,9 +122,9 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #PATH_TO_MODEL = 'models/best_YOLOv11n_1280_real_half.onnx'
     PATH_TO_MODEL = 'models/best_YOLOv8n.pt'
     INFERENCE_ENGINE_TYPE = 'pt'
-    INPUT_SIZE = 1280
+    INPUT_SIZE = 640
     N_TEST_BATCHES = 2
-    BATCH_SIZE = 128 # Can be adjusted as needed
+    BATCH_SIZE = 8 # Can be adjusted as needed
 
     def preprocessor(frame):
         #frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # Only when read from file