Spaces:
Sleeping
Sleeping
Commit
·
0af87d2
1
Parent(s):
be5e5b5
accepting input from upload or mic
Browse files
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 |
-
|
14 |
elif audio is not None:
|
15 |
-
|
16 |
else:
|
17 |
return "(please provide audio)"
|
18 |
|
19 |
-
audio = whisper.load_audio(
|
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)
|