add code

app/main.py

@@ -1,5 +1,4 @@
-from tensorflow.keras.models import load_model as keras_load_model
-
+import keras
 from fastapi import FastAPI, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 import numpy as np
@@ -35,7 +34,7 @@ def initialize_model():
                 local_dir=HF_MODEL_DIR,
                 local_dir_use_symlinks=False
             )
-            MODEL = keras_load_model(model_path)
+            MODEL = keras.saving.load_model(model_path)
             logging.info("✅ Model loaded successfully.")
         except Exception as e:
             logging.error(f"❌ Failed to load model: {e}")

requirements.txt

@@ -3,5 +3,6 @@ uvicorn
 numpy
 pillow
 tensorflow-cpu
+keras==3.5.0
 huggingface_hub
 python-multipart