Update app.py

app.py

@@ -2,12 +2,29 @@ import gradio as gr
 from ultralytics import YOLO
 from PIL import Image, ImageOps, ImageEnhance
 import numpy as np
-import tempfile, io, base64
+import io, base64
 
-# Load models
-model_swelling = YOLO("models/swelling/swelling_final.pt")
-model_redness = YOLO("models/redness/redness_final.pt")
-model_bleeding = YOLO("models/bleeding/bleeding_final.pt")
+# =========================================================
+# Lazy-loaded global models (LOAD ONLY ON FIRST REQUEST)
+# =========================================================
+model_swelling = None
+model_redness = None
+model_bleeding = None
+
+def get_models():
+    """Load YOLO models only once (lazy loading)."""
+    global model_swelling, model_redness, model_bleeding
+
+    if model_swelling is None:
+        model_swelling = YOLO("models/swelling/swelling_final.pt")
+
+    if model_redness is None:
+        model_redness = YOLO("models/redness/redness_final.pt")
+
+    if model_bleeding is None:
+        model_bleeding = YOLO("models/bleeding/bleeding_final.pt")
+
+    return model_swelling, model_redness, model_bleeding
 
 
 # --- Helpers ---
@@ -49,13 +66,16 @@ def detect_gingivitis(image, conf=0.25, iou=0.5):
         if image is None:
             return [None, None, None, "❌ No image uploaded"]
 
+        # Load models (only once)
+        sw_model, rd_model, bl_model = get_models()
+
         # Preprocess image
         image = preprocess(image)
 
-        # Run models
-        sw_res = model_swelling.predict(image, conf=conf, iou=iou)
-        rd_res = model_redness.predict(image, conf=conf, iou=iou)
-        bl_res = model_bleeding.predict(image, conf=conf, iou=iou)
+        # Run detections
+        sw_res = sw_model.predict(image, conf=conf, iou=iou)
+        rd_res = rd_model.predict(image, conf=conf, iou=iou)
+        bl_res = bl_model.predict(image, conf=conf, iou=iou)
 
         # Convert to images
         img_sw = sw_res[0].plot(labels=False)[:, :, ::-1]
@@ -105,6 +125,13 @@ interface = gr.Interface(
     title="Gingivitis Detection"
 )
 
+# =========================================================
+# Warm-start: preload models on startup
+# =========================================================
+print("🔥 Preloading models to reduce Render cold start...")
+get_models()
+print("✅ Gingivitis models ready")
+
 # interface.launch()
 if __name__ == "__main__":
     interface.launch(server_name="0.0.0.0", server_port=7860, show_error=True)