chinmaydan commited on
Commit
0af87d2
·
1 Parent(s): be5e5b5

accepting input from upload or mic

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,13 +10,13 @@ model = whisper.load_model("small")
10
  def predict(audio, mic_audio=None):
11
  # audio = tuple (sample_rate, frames) or (sample_rate, (frames, channels))
12
  if mic_audio is not None:
13
- sampling_rate, waveform = mic_audio
14
  elif audio is not None:
15
- sampling_rate, waveform = audio
16
  else:
17
  return "(please provide audio)"
18
 
19
- audio = whisper.load_audio(waveform)
20
  audio = whisper.pad_or_trim(audio)
21
 
22
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
 
10
  def predict(audio, mic_audio=None):
11
  # audio = tuple (sample_rate, frames) or (sample_rate, (frames, channels))
12
  if mic_audio is not None:
13
+ input_audio = mic_audio
14
  elif audio is not None:
15
+ input_audio = audio
16
  else:
17
  return "(please provide audio)"
18
 
19
+ audio = whisper.load_audio(input_audio)
20
  audio = whisper.pad_or_trim(audio)
21
 
22
  mel = whisper.log_mel_spectrogram(audio).to(model.device)