softwareproj / app.py
solie79's picture
Update app.py
dd1926a verified
raw
history blame
882 Bytes
import gradio as gr
from transformers import pipeline
from gtts import gTTS
# Load translation model
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
def translate_and_speak(text, target_language):
# Translate text
translated_text = translator(text, tgt_lang=target_language)[0]['translation_text']
# Convert text to speech
tts = gTTS(translated_text, lang=target_language)
output_path = "output.mp3"
tts.save(output_path)
return translated_text, output_path
# Gradio Interface
demo = gr.Interface(
fn=translate_and_speak,
inputs=[
gr.Textbox(label="Enter Text"),
gr.Dropdown(["fr", "es", "de", "it", "zh", "ar", "hi", "ur"], label="Target Language")
],
outputs=[
gr.Textbox(label="Translated Text"),
gr.Audio(label="Generated Speech")
]
)
demo.launch(share=True)