Chanjeans commited on
Commit
feae2b6
ยท
verified ยท
1 Parent(s): f692eaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -6,15 +6,21 @@ import numpy as np
6
  from fastapi import FastAPI, HTTPException
7
  from pydantic import BaseModel
8
  from sentence_transformers import SentenceTransformer
 
 
 
9
 
10
  #####################################
11
  # 1) ์•ฑ ๋ฐ ๋ชจ๋ธ ์ดˆ๊ธฐํ™”
12
  #####################################
13
  app = FastAPI()
14
  os.environ["HF_HOME"] = "/tmp/huggingface"
 
 
 
15
  # ๐ŸŸข (A) ๋ชจ๋ธ ๋กœ๋”ฉ
16
- print("Loading SentenceTransformer model... (msmarco-distilbert-base-v4)")
17
- model_bert = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
18
  print("Model loaded.")
19
 
20
  # ๐ŸŸข (B) ์˜ˆ์‹œ ์•„์ดํ…œ ๋ชฉ๋ก
 
6
  from fastapi import FastAPI, HTTPException
7
  from pydantic import BaseModel
8
  from sentence_transformers import SentenceTransformer
9
+ import os
10
+
11
+
12
 
13
  #####################################
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
  # ๐ŸŸข (A) ๋ชจ๋ธ ๋กœ๋”ฉ
22
+ print("Loading SentenceTransformer model...)")
23
+ model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", trust_remote_code=True)
24
  print("Model loaded.")
25
 
26
  # ๐ŸŸข (B) ์˜ˆ์‹œ ์•„์ดํ…œ ๋ชฉ๋ก