Update main.py

main.py

@@ -134,20 +134,28 @@ else:
 #     )
 from huggingface_hub import snapshot_download
 
-HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+HF_TOKEN   = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+CACHE_DIR  = os.getenv("HF_HOME", "/app/.cache")
+LOCAL_DERM = os.getenv("DERM_LOCAL_DIR", "/app/derm-foundation")  # จะดาวน์โหลดมาไว้ที่นี่
+
+os.makedirs(CACHE_DIR, exist_ok=True)
+os.makedirs(LOCAL_DERM, exist_ok=True)
 
 logger.info("Loading Derm Foundation (first time may take a while)...")
 try:
-    if DERM_LOCAL_DIR and os.path.isdir(DERM_LOCAL_DIR) and os.path.exists(os.path.join(DERM_LOCAL_DIR, "saved_model.pb")):
-        derm_dir = DERM_LOCAL_DIR
-        logger.info(f"Loaded Derm Foundation from local: {DERM_LOCAL_DIR}")
+    if os.path.isdir(LOCAL_DERM) and os.path.exists(os.path.join(LOCAL_DERM, "saved_model.pb")):
+        derm_dir = LOCAL_DERM
+        logger.info(f"Loaded Derm Foundation from local: {derm_dir}")
     else:
         logger.info(f"Downloading derm-foundation from hub: {DERM_MODEL_ID}")
         derm_dir = snapshot_download(
             repo_id=DERM_MODEL_ID,
             repo_type="model",
             allow_patterns=["saved_model.pb", "variables/*"],
-            token=HF_TOKEN,                 # ← เพิ่มบรรทัดนี้
+            token=HF_TOKEN,                     # <-- ต้องมี
+            cache_dir=CACHE_DIR,                # <-- ไม่เขียนไป /.cache
+            local_dir=LOCAL_DERM,               # <-- เก็บเป็นโฟลเดอร์จริง
+            local_dir_use_symlinks=False,       # <-- กันปัญหา symlink/สิทธิ์
         )
         logger.info(f"Derm Foundation downloaded to: {derm_dir}")
 
@@ -159,6 +167,7 @@ except Exception as e:
         "Make sure you accepted the model terms and set HF_TOKEN in Space Settings."
     )
 
+
 # ---------------------- Utils ----------------------
 def pil_to_png_bytes_448(pil_img: Image.Image) -> bytes:
     pil_img = pil_img.convert("RGB").resize(DF_SIZE)