saima730's picture
Update app.py
ca40be0 verified
raw
history blame
1.81 kB
st.write(f"**Translated text in {target_language}:**")
st.write(translated_text)
st.write(f"""**Translated text in {target_language}:**""")
st.write(translated_text)
import streamlit as st
from transformers import pipeline
# Load the translation pipeline from Hugging Face
@st.cache_resource
def load_translation_pipeline(model_name="Helsinki-NLP/opus-mt-en-ROMANCE"):
return pipeline("translation", model=model_name)
# Function to translate text
def translate_text(translation_pipeline, text, target_language):
# Modify the model name to reflect the target language
if target_language == "French":
translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-fr"
elif target_language == "Spanish":
translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-es"
elif target_language == "German":
translation_pipeline.model_name = "Helsinki-NLP/opus-mt-en-de"
else:
st.error("Target language not supported!")
return None
translation = translation_pipeline(text)
return translation[0]['translation_text']
# Streamlit app layout
st.title("Language Translator")
# Input text to translate
text = st.text_area("Enter text in English to translate:")
# Select target language
target_language = st.selectbox(
"Select target language:",
["French", "Spanish", "German"] # Add more languages if needed
)
# Load the translation pipeline
translation_pipeline = load_translation_pipeline()
# Translate button
if st.button("Translate"):
if text:
translated_text = translate_text(translation_pipeline, text, target_language)
if translated_text:
st.write(f"**Translated text in {target_language}:**")
st.write(translated_text)
else:
st.error("Please enter text to translate.")