salmanmapkar commited on
Commit
3cdba3e
1 Parent(s): cc28378

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ import moviepy.editor as mp
17
  import datetime
18
  import torch
19
  import pyannote.audio
20
- from pyannote.audio.pipelines.speaker_verification import NeMoPretrainedSpeakerEmbedding
21
  from pyannote.audio import Audio
22
  from pyannote.core import Segment
23
  import wave
@@ -141,7 +141,7 @@ def Transcribe_V1(NumberOfSpeakers, SpeakerNames="", audio="temp_audio.wav"):
141
 
142
  def Transcribe_V2(num_speakers, speaker_names, audio="temp_audio.wav"):
143
  model = whisper.load_model("medium")
144
- embedding_model = NeMoPretrainedSpeakerEmbedding()
145
  # embedding_model = PretrainedSpeakerEmbedding(
146
  # "speechbrain/spkrec-ecapa-voxceleb",
147
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
17
  import datetime
18
  import torch
19
  import pyannote.audio
20
+ from pyannote.audio.pipelines.speaker_verification import PyannoteAudioPretrainedSpeakerEmbedding
21
  from pyannote.audio import Audio
22
  from pyannote.core import Segment
23
  import wave
 
141
 
142
  def Transcribe_V2(num_speakers, speaker_names, audio="temp_audio.wav"):
143
  model = whisper.load_model("medium")
144
+ embedding_model = PyannoteAudioPretrainedSpeakerEmbedding("pyannote/embedding")
145
  # embedding_model = PretrainedSpeakerEmbedding(
146
  # "speechbrain/spkrec-ecapa-voxceleb",
147
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')