Spaces:
Build error
Build error
File size: 1,761 Bytes
8f8ee33 08c575e 8f8ee33 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import gradio as gr
import whisper
from pybraille import convertText
# Translate speech to braille
model = whisper.load_model('base')
def speech_to_text(audio):
"""
:param audio: Speech data
:return: Transcript of speech data
"""
return model.transcribe(audio)
def transcribe_audio(audio, store=None):
"""
:param audio: Speech data from gradio interface
:param store: Container for user data
:return: Detected language, transcript text, braille translation, session state
"""
if store is None:
store = {}
result = speech_to_text(audio)
store['transcription'] = result['text']
store['braille'] = convertText(store['transcription'])
return f"Detected language: {result['language']}", store[
'transcription'], store['braille'], store
# Setup gradio interface
title = "Speech to Braille Translator"
transcription_tb = gr.Textbox(label="Transcription", lines=10, max_lines=20)
detected_lang = gr.outputs.HTML(label="Detected Language")
transcription_braille = gr.Textbox(label="Braille", lines=10, max_lines=20)
state = gr.State({"transcription": ""})
demo = gr.Interface(fn=transcribe_audio,
inputs=[
gr.Audio(source="microphone", type="filepath",
streaming=False),
state
],
outputs=[
detected_lang,
transcription_tb,
transcription_braille,
state,
],
# live=True,
allow_flagging='never',
title=title,
)
# Launch gradio app
demo.launch()
|