Update app.py

app.py

@@ -24,8 +24,15 @@ app.add_middleware(
 labels = ['butterfly', 'cow', 'dog', 'elephant', 'goat', 'hen', 'horse', 'spyder']
 
 # 🔥 load model (Keras 3 fix)
-model = TFSMLayer("saved_model", call_endpoint="serving_default")
-
+model = None
+
+def get_model():
+    global model
+    if model is None:
+        print("🔥 Loading model...")
+        model = TFSMLayer("saved_model", call_endpoint="serving_default")
+    return model
+    
 def preprocess(image):
     image = image.resize((224, 224))
     image = np.array(image).astype("float32")
@@ -46,7 +53,7 @@ async def predict(file: UploadFile = File(...)):
         input_data = preprocess(image)
 
         # 🔥 inference
-        pred = model(input_data)
+        pred = get_model()(input_data)
 
         # handle dict output
         if isinstance(pred, dict):