Spaces:
Runtime error
Runtime error
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):
|