rein0421 commited on
Commit
8fa0d1d
·
verified ·
1 Parent(s): 12b8b38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -6,10 +6,12 @@ import numpy as np
6
  from pyannote.audio import Model, Inference
7
  from pydub import AudioSegment
8
 
9
- os.environ["HUGGINGFACE_HUB_TOKEN"] = os.environ.get("HF") # トークンを適切に設定
 
 
10
 
11
- # 事前学習済みモデルの読み込み
12
- model = Model.from_pretrained("pyannote/embedding")
13
  inference = Inference(model)
14
 
15
  def cosine_similarity(vec1, vec2):
 
6
  from pyannote.audio import Model, Inference
7
  from pydub import AudioSegment
8
 
9
+ hf_token = os.environ.get("HUGGINGFACE_HUB_TOKEN")
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):