subtitle_gen / app.py
robinmia's picture
Update app.py
f27c192 verified
raw
history blame contribute delete
No virus
318 Bytes
import whisper
import gradio as gr
def getSubTitle(audio_file):
model = whisper.load_model("base")
audio = whisper.load_audio(audio_file)
result = model.transcribe(audio)['segments']
return result
demo = gr.Interface(fn=getSubTitle, inputs=gr.Audio(type="filepath"), outputs=["text"])
demo.launch()