sssssungk commited on
Commit
a887a28
·
verified ·
1 Parent(s): c911c9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -134,7 +134,8 @@ def real_fake_check(list_dir, path, model):
134
 
135
 
136
  def main(file_name):
137
- pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
 
138
  #device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
139
  device = torch.device('cpu')
140
 
 
134
 
135
 
136
  def main(file_name):
137
+ hf_token = os.getenv("HUGGINGFACE_TOKEN")
138
+ pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1",use_auth_token=hf_token)
139
  #device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
140
  device = torch.device('cpu')
141