Update app.py

app.py

@@ -6,6 +6,7 @@ import numpy as np
 import cv2
 import requests
 from fastapi import FastAPI, File, UploadFile, HTTPException
+from huggingface_hub import hf_hub_download  # NEW
 
 os.environ["KERAS_BACKEND"] = "jax"
 import keras
@@ -17,7 +18,15 @@ IMG_SIZE = 256
 THRESHOLD = 0.05  # ideally derived from healthy images [file:1]
 
 # ─── Load model once at startup ───────────────────────────────────────────────
-MODEL = keras.saving.load_model("hf://sairamv1/TeethCariesDetection_CNNAutoencoder")
+MODEL_REPO = "sairamv1/TeethCariesDetection_CNNAutoencoder"
+MODEL_FILENAME = "autoencoder.keras"  # or whatever file you uploaded
+
+local_model_path = hf_hub_download(
+    repo_id=MODEL_REPO,
+    filename=MODEL_FILENAME,
+)
+
+MODEL = keras.saving.load_model(local_model_path)
 
 app = FastAPI()