subtitle_gen / app.py
robinmia's picture
a
c8b1a0f verified
raw
history blame
No virus
337 Bytes
import whisper
import gradio as gr
def getSubTitle(audio_file):
import whisper
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()