salmanmapkar commited on
Commit
2f9b63d
1 Parent(s): f20220f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -103,7 +103,7 @@ def Transcribe_V1(NumberOfSpeakers, SpeakerNames="", audio="temp_audio.wav"):
103
  return f"dz_{audio}.wav", dzList, segments
104
 
105
  def transcribe(dz_audio):
106
- model = whisper.load_model("large-v2")
107
  result = model.transcribe(dz_audio)
108
  # for _ in result['segments']:
109
  # print(_['start'], _['end'], _['text'])
@@ -138,7 +138,7 @@ def Transcribe_V1(NumberOfSpeakers, SpeakerNames="", audio="temp_audio.wav"):
138
 
139
 
140
  def Transcribe_V2(num_speakers, speaker_names, audio="temp_audio.wav"):
141
- model = whisper.load_model("large-v2")
142
  embedding_model = PretrainedSpeakerEmbedding(
143
  "speechbrain/spkrec-ecapa-voxceleb",
144
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
103
  return f"dz_{audio}.wav", dzList, segments
104
 
105
  def transcribe(dz_audio):
106
+ model = whisper.load_model("medium")
107
  result = model.transcribe(dz_audio)
108
  # for _ in result['segments']:
109
  # print(_['start'], _['end'], _['text'])
 
138
 
139
 
140
  def Transcribe_V2(num_speakers, speaker_names, audio="temp_audio.wav"):
141
+ model = whisper.load_model("medium")
142
  embedding_model = PretrainedSpeakerEmbedding(
143
  "speechbrain/spkrec-ecapa-voxceleb",
144
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')