Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline | |
# Available languages and their corresponding models | |
available_languages = { | |
"French": "Helsinki-NLP/opus-mt-en-fr", | |
"German": "Helsinki-NLP/opus-mt-en-de", | |
"Spanish": "Helsinki-NLP/opus-mt-en-es", | |
"Chinese": "Helsinki-NLP/opus-mt-en-zh", | |
"Japanese": "Helsinki-NLP/opus-mt-en-jap", | |
"Russian": "Helsinki-NLP/opus-mt-en-ru", | |
"Arabic": "Helsinki-NLP/opus-mt-en-ar", | |
"Urdu": "Helsinki-NLP/opus-mt-en-ur", | |
} | |
# Streamlit app title | |
st.title("Language Translator") | |
# User input for text to translate | |
text_to_translate = st.text_area("Enter text in English:", "") | |
# Language selection | |
target_language = st.selectbox("Select the target language:", list(available_languages.keys())) | |
# Load the translation model based on the selected language | |
translator = pipeline("translation", model=available_languages[target_language]) | |
# Translate button | |
if st.button("Translate"): | |
if text_to_translate: | |
# Perform the translation | |
translation = translator(text_to_translate) | |
# Display the translated text | |
st.write(f"**Translated text in {target_language}:**") | |
st.write(translation[0]['translation_text']) | |
else: | |
st.warning("Please enter some text to translate.") | |
# Footer | |
st.markdown("Powered by [Hugging Face Transformers](https://huggingface.co/transformers/).") | |