Spaces:
Build error
Build error
Commit
·
89f5759
1
Parent(s):
5026395
Update app.py
Browse files
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()
|
|