rein0421 commited on
Commit
1cfebfc
·
verified ·
1 Parent(s): fd9ed96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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
- # use_auth_token を明示的に渡す
14
- model = Model.from_pretrained("pyannote/embedding", use_auth_token=hf_token)
 
 
 
 
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):