Fix HF runtime model deserialization compatibility

Dockerfile

@@ -1,11 +1,13 @@
-FROM python:3.10-slim
+FROM python:3.12-slim
 WORKDIR /app
 
 COPY ./requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
 COPY . .
 
+ENV PYTHONUNBUFFERED=1
+
 EXPOSE 7860
 
 # Lệnh khởi chạy API FastAPI (Hugging Face dùng port 7860)

main.py

@@ -1,3 +1,4 @@
+import os
 from io import BytesIO
 from PIL import Image
 import tensorflow as tf
@@ -44,7 +45,32 @@ DISEASE_INFO = {
     }
 }
 
-model = load_model('plant_disease_recog_model_pwp_3.keras')
+MODEL_PATH = os.getenv("MODEL_PATH", "plant_disease_recog_model_pwp_3.keras")
+
+
+def load_model_with_compatibility(model_path):
+    # Some Keras/TensorFlow versions do not accept this key during deserialization.
+    from tensorflow.keras.layers import Dense
+
+    original_dense_init = Dense.__init__
+
+    def patched_dense_init(self, *args, **kwargs):
+        kwargs.pop("quantization_config", None)
+        return original_dense_init(self, *args, **kwargs)
+
+    Dense.__init__ = patched_dense_init
+
+    try:
+        try:
+            return load_model(model_path, compile=False, safe_mode=False)
+        except TypeError:
+            return load_model(model_path, compile=False)
+    finally:
+        Dense.__init__ = original_dense_init
+
+
+model = load_model_with_compatibility(MODEL_PATH)
+print(f"[Startup] TensorFlow={tf.__version__}, model={MODEL_PATH}")
 
 def read_file_as_image(data):
     img = Image.open(BytesIO(data))
@@ -52,6 +78,16 @@ def read_file_as_image(data):
     img = img.resize((224, 224))
     return np.array(img)
 
+
+@app.get("/")
+async def root():
+    return {"status": "ok", "message": "SmartFarm disease API is running"}
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok", "model_loaded": model is not None}
+
 @app.post("/detect-disease")
 async def detectDisease(file: UploadFile = File(...)):
 

requirements.txt

@@ -1,6 +1,7 @@
-fastapi
-uvicorn
-python-multipart
-tensorflow
-pillow
-numpy
+fastapi==0.136.0
+uvicorn==0.44.0
+python-multipart==0.0.20
+tensorflow==2.21.0
+keras==3.14.0
+pillow==12.2.0
+numpy>=1.26.0,<2.2.0