AI_TEST / app.py
MusIre's picture
Update app.py
3407e42
raw
history blame
759 Bytes
import subprocess
subprocess.run(["pip", "install", "-U", "pip"])
subprocess.run(["pip", "install", "-U", "gradio"])
subprocess.run(["pip", "install", "whisper"])
import gradio as gr
import whisper
def transcribe_audio(audio_file):
model = whisper.load_model("base")
result = model.transcribe(audio_file)
return result["text"]
def main():
audio_input = gr.Audio(type="file")
output_text = gr.Textbox()
iface = gr.Interface(fn=transcribe_audio, inputs=audio_input,
outputs=output_text, title="Audio Transcription App",
description="Upload an audio file and hit the 'Submit'\
button")
iface.launch()
if __name__ == '__main__':
main()