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()