Spaces:
Sleeping
Sleeping
import streamlit as st | |
import time | |
from transformers import pepeline | |
st.title("Traductor multilenguaje") | |
translation_models = { | |
"English to German": "Helsinki-NLP/opus-mt-en-de", | |
"German to English": "Helsinki-NLP/opus-mt-de-en", | |
"English to French": "Helsinki-NLP/opus-mt-en-fr", | |
"French to English": "Helsinki-NLP/opus-mt-fr-en", | |
"English to Urdu": "Helsinki-NLP/opus-mt-en-ur", | |
"Urdu to English": "Helsinki-NLP/opus-mt-ur-en", | |
"English to Spanish": "Helsinki-NLP/opus-mt-en-es", | |
"Spanish to English": "Helsinki-NLP/opus-mt-es-en", | |
"English to Chinese": "Helsinki-NLP/opus-mt-en-zh", | |
"Chinese to English": "Helsinki-NLP/opus-mt-zh-en", | |
} | |
idiomaseleccionado = st.selectbox("Idiomas:" , list(translation_models.keys())) | |
traductor = pipeline(task="translation", model= translation_models[idiomaseleccionado]) | |
textoingresado = st.text_area("Ingrese el texto a traducir:","") | |
if st.button("Traducir"): | |
with st.spinner("Traduciendo..."): | |
time.sleep(2) | |
if user_input: | |
textotraducido= traductor(user_input, max_length=500)[0]["textotraducido"] | |
st.succes("Texto Traducido:" {textotraducido}) | |
else: | |
st.warning("Ingrese un texto") | |
if st.button("Limpiar"): | |
textoingresado="" | |
st.empty() | |