Chanjeans commited on
Commit
7ae68e3
·
verified ·
1 Parent(s): 3be00cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -14,15 +14,15 @@ import os
14
  # 1) 앱 및 모델 초기화
15
  #####################################
16
  app = FastAPI()
17
- os.environ["HF_HOME"] = "/tmp/huggingface"
18
- os.environ["HF_HOME"] = "/tmp/huggingface_cache"
19
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
20
- os.environ["HF_HUB_CACHE"] = "/tmp/huggingface_cache"
21
- os.environ["HF_HUB_OFFLINE"] = "1" # 모델 다운로드 없이 로드
22
- os.environ["TRANSFORMERS_OFFLINE"] = "1" # Transformers도 오프라인 모드로
23
  # 🟢 (A) 모델 로딩
24
  print("Loading SentenceTransformer model...)")
25
- model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", trust_remote_code=False)
26
  print("Model loaded.")
27
 
28
  # 🟢 (B) 예시 아이템 목록
 
14
  # 1) 앱 및 모델 초기화
15
  #####################################
16
  app = FastAPI()
17
+ import os
18
+
19
+ os.environ["HF_HOME"] = "/workspace/huggingface_cache"
20
+ os.environ["HF_HUB_CACHE"] = "/workspace/huggingface_cache"
21
+ os.environ["TRANSFORMERS_CACHE"] = "/workspace/huggingface_cache"
22
+
23
  # 🟢 (A) 모델 로딩
24
  print("Loading SentenceTransformer model...)")
25
+ model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
26
  print("Model loaded.")
27
 
28
  # 🟢 (B) 예시 아이템 목록