shuaige / app.py
wang0507's picture
Update app.py
c6ec7e0
raw
history blame
795 Bytes
import os
os.system("pip install git+https://github.com/openai/whisper.git")
import gradio as gr
import whisper
model = whisper.load_model("large")
def inference(audio):
audio = whisper.load_audio(audio)
audio = whisper.pad_or_trim(audio)
mel = whisper.log_mel_spectrogram(audio).to(model.device)
_, probs = model.detect_language(mel)
options = whisper.DecodingOptions(fp16 = False)
result = whisper.decode(model, mel, options)
print(result.text)
return result.text, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
iface = gr.Interface(
fn=inference,
inputs=gr.Audio(type="filepath", label="上传音频文件 (.mp3, .wav等)"),
outputs="text"
)
# 启动 Gradio 界面
iface.launch()