gpenessot's picture
Create app.py
8baaf83
import gradio as gr
from transformers import pipeline
# Load the translation pipeline from Hugging Face Transformers
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-en")
# Define the translation function
def translate_text(text: str) -> str:
"""Translate text from French to English
Args:
text (str): french text input
Returns:
str: english text output
"""
try:
# Translate text from French to English
translation_result = translator(text,
src_lang="fr",
tgt_lang="en")
translated_text = translation_result[0]['translation_text']
return translated_text
except Exception as e:
# Print the error for debugging
print(f"Translation Error: {str(e)}")
return "Translation Error"
# Create a "simple" Gradio interface
interface = gr.Interface(
fn=translate_text,
inputs=gr.Textbox("text", label="Enter text in French"),
outputs='text',#gr.Textbox("text", label="Translated text in English"),
#live=True,
title="Language Translator",
description="Translate French text to English using Hugging Face Transformers.",
)
# Launch the Gradio interface
interface.launch()