Update main.py

main.py

@@ -110,6 +110,32 @@ else:
     best_th = np.full(C, 0.5, dtype="float32")
 
 # ---------------------- Load derm-foundation ----------------------
+# 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}")
+#     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/*"],
+#         )
+#         logger.info(f"Derm Foundation downloaded to: {derm_dir}")
+
+#     derm = tf.saved_model.load(derm_dir)
+#     infer = derm.signatures["serving_default"]  # call with key 'inputs'
+# except Exception as e:
+#     raise RuntimeError(
+#         f"Failed to load derm-foundation: {e}. "
+#         "Make sure you accepted the model terms on Hugging Face, "
+#         "or set DERM_LOCAL_DIR to a local SavedModel path."
+#     )
+from huggingface_hub import snapshot_download
+
+HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+
 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")):
@@ -121,16 +147,16 @@ try:
             repo_id=DERM_MODEL_ID,
             repo_type="model",
             allow_patterns=["saved_model.pb", "variables/*"],
+            token=HF_TOKEN,                 # ← เพิ่มบรรทัดนี้
         )
         logger.info(f"Derm Foundation downloaded to: {derm_dir}")
 
     derm = tf.saved_model.load(derm_dir)
-    infer = derm.signatures["serving_default"]  # call with key 'inputs'
+    infer = derm.signatures["serving_default"]
 except Exception as e:
     raise RuntimeError(
         f"Failed to load derm-foundation: {e}. "
-        "Make sure you accepted the model terms on Hugging Face, "
-        "or set DERM_LOCAL_DIR to a local SavedModel path."
+        "Make sure you accepted the model terms and set HF_TOKEN in Space Settings."
     )
 
 # ---------------------- Utils ----------------------