on1onmangoes commited on
Commit
7a90588
·
verified ·
1 Parent(s): 7bb0efa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,8 +20,9 @@ from transformers.pipelines.audio_utils import ffmpeg_read
20
  HF_TOKEN = os.environ.get("HF_TOKEN")
21
 
22
  # set up the diarization pipeline
23
- diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.0", use_auth_token=HF_TOKEN)
24
  #diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=HF_TOKEN)
 
25
 
26
  if torch.cuda.is_available():
27
  diarization_pipeline.to(torch.device("cuda"))
 
20
  HF_TOKEN = os.environ.get("HF_TOKEN")
21
 
22
  # set up the diarization pipeline
23
+ #diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.0", use_auth_token=HF_TOKEN)
24
  #diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=HF_TOKEN)
25
+ diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1, use_auth_token=HF_TOKEN)
26
 
27
  if torch.cuda.is_available():
28
  diarization_pipeline.to(torch.device("cuda"))