|
import streamlit as st |
|
from transformers import pipeline |
|
|
|
|
|
st.title("Translator App πΉ") |
|
st.write("Translate text between different languages using AI.") |
|
|
|
|
|
language_models = { |
|
"English β French": "Helsinki-NLP/opus-mt-en-fr", |
|
"French β English": "Helsinki-NLP/opus-mt-fr-en", |
|
"English β Spanish": "Helsinki-NLP/opus-mt-en-es", |
|
"Spanish β English": "Helsinki-NLP/opus-mt-es-en", |
|
} |
|
|
|
|
|
selected_model = st.selectbox("Select Language Pair", list(language_models.keys())) |
|
|
|
|
|
translator = pipeline("translation", model=language_models[selected_model]) |
|
|
|
|
|
text = st.text_area("Enter text to translate:") |
|
|
|
|
|
if st.button("Translate"): |
|
if text: |
|
translated_text = translator(text)[0]["translation_text"] |
|
st.success(f"**Translated Text:** {translated_text}") |
|
else: |
|
st.warning("β οΈ Please enter text to translate!") |
|
|
|
|
|
st.caption("πΉ Powered by Hugging Face Transformers & Streamlit") |