Upload app.py

app.py

@@ -8,7 +8,8 @@ import os
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-MODEL_ID = os.environ.get("HF_MODEL_ID", "")
+MODEL_ID = (os.environ.get("HF_MODEL_ID") or "").strip()
+HF_TOKEN = (os.environ.get("HF_TOKEN") or "").strip() or None  # 비공개 리포일 때만 설정
 model = None
 
 app = FastAPI(title="Reranker API")
@@ -22,7 +23,8 @@ def load_model():
         raise RuntimeError("Space 설정에서 HF_MODEL_ID 를 지정하세요 (예: username/certweb-reranker)")
     from sentence_transformers import CrossEncoder
     print(f"모델 로딩 중: {MODEL_ID}")
-    model = CrossEncoder(MODEL_ID)
+    # 비공개 리포면 HF_TOKEN (Secrets) 설정 필요
+    model = CrossEncoder(MODEL_ID, token=HF_TOKEN)
     print("로드 완료.")