Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,12 @@ hf_token = os.environ.get("HF")
|
|
10 |
if hf_token is None:
|
11 |
raise ValueError("HUGGINGFACE_HUB_TOKEN が設定されていません。")
|
12 |
|
13 |
-
#
|
14 |
-
|
|
|
|
|
|
|
|
|
15 |
inference = Inference(model)
|
16 |
|
17 |
def cosine_similarity(vec1, vec2):
|
|
|
10 |
if hf_token is None:
|
11 |
raise ValueError("HUGGINGFACE_HUB_TOKEN が設定されていません。")
|
12 |
|
13 |
+
# 書き込み可能なキャッシュディレクトリを指定
|
14 |
+
cache_dir = "/tmp/hf_cache"
|
15 |
+
os.makedirs(cache_dir, exist_ok=True)
|
16 |
+
|
17 |
+
# use_auth_token と cache_dir を指定してモデルを読み込む
|
18 |
+
model = Model.from_pretrained("pyannote/embedding", use_auth_token=hf_token, cache_dir=cache_dir)
|
19 |
inference = Inference(model)
|
20 |
|
21 |
def cosine_similarity(vec1, vec2):
|