robinmia/subtitle_gen
0
1import whisper2import gradio as gr3 4def getSubTitle(audio_file):5 model = whisper.load_model("base")6 audio = whisper.load_audio(audio_file)7 result = model.transcribe(audio)['segments']8 return result9 10demo = gr.Interface(fn=getSubTitle, inputs=gr.Audio(type="filepath"), outputs=["text"])11 12demo.launch()13 