fix: align ML service labels with training order

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
Asep Haryana Saputra
2026-05-22 19:40:41 +00:00
co-authored by Claude Opus 4.7
parent 59877bc17f
commit ed8cecc3f5
3 changed files with 21 additions and 1 deletions
+3 -1
View File
@@ -1,4 +1,5 @@
from io import BytesIO
import logging
import os
from pathlib import Path
@@ -7,7 +8,7 @@ from PIL import Image, UnidentifiedImageError
import tensorflow as tf
LABELS = ["Bercak Daun", "Hawar Daun", "Karat Daun", "Daun Sehat"]
LABELS = ["Bercak Daun", "Daun Sehat", "Karat Daun", "Hawar Daun"]
SERVICE_NAME = "zeavis-ml-service"
SERVICE_VERSION = "0.1.0"
@@ -40,6 +41,7 @@ class ModelService:
except Exception as exc:
self.model = None
self.load_error = str(exc)
logging.exception("Failed to load ML model from %s", self.model_path)
def preprocess(self, image_bytes: bytes) -> np.ndarray:
try: