add code

app/main.py

@@ -1,4 +1,5 @@
-from tensorflow.keras.models import load_model
+from tensorflow.keras.models import load_model as keras_load_model
+
 from fastapi import FastAPI, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 import numpy as np
@@ -22,25 +23,23 @@ app.add_middleware(
 
 MODEL = None 
 
-def load_model():
+def initialize_model():
     global MODEL
     if MODEL is None:
         HF_MODEL_DIR = "/tmp/rice_model"
         os.makedirs(HF_MODEL_DIR, exist_ok=True)
         try:
-         
             model_path = hf_hub_download(
                 repo_id="ShahzadAli44/rice_cnn",
                 filename="rice_cnn_model.keras",
                 local_dir=HF_MODEL_DIR,
-                local_dir_use_symlinks=False  
+                local_dir_use_symlinks=False
             )
-            MODEL = load_model(model_path)
+            MODEL = keras_load_model(model_path)
             logging.info("✅ Model loaded successfully.")
         except Exception as e:
             logging.error(f"❌ Failed to load model: {e}")
 
-
 CLASS_NAMES = [
     "bacterial_leaf_blight", "brown_spot", "healthy", "leaf_blast",
     "leaf_scald", "narrow_brown_spot", "rice_hispa", "sheath_blight", "tungro"
@@ -102,7 +101,7 @@ def home():
 
 @app.post("/predict")
 async def predict(file: UploadFile = File(...)):
-    load_model()  
+    initialize_model() 
     if MODEL is None:
         return {"error": "Model failed to load."}