frogcho123 commited on
Commit
89f5759
1 Parent(s): 5026395

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import gradio as gr
2
  import whisper
 
3
 
4
  model = whisper.load_model("base")
5
 
6
  def transcribe(audio):
 
 
7
  audio = whisper.pad_or_trim(audio)
8
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
9
  options = whisper.DecodingOptions()
@@ -12,4 +15,3 @@ def transcribe(audio):
12
 
13
  iface = gr.Interface(fn=transcribe, inputs="audio", outputs="text")
14
  iface.launch()
15
-
 
1
  import gradio as gr
2
  import whisper
3
+ import numpy as np
4
 
5
  model = whisper.load_model("base")
6
 
7
  def transcribe(audio):
8
+ if not isinstance(audio, np.ndarray):
9
+ audio = np.array(audio)
10
  audio = whisper.pad_or_trim(audio)
11
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
12
  options = whisper.DecodingOptions()
 
15
 
16
  iface = gr.Interface(fn=transcribe, inputs="audio", outputs="text")
17
  iface.launch()