CoolFace
Apppublic

ritampatra/Translator_PROO

sourceHugging Facemitupdated 2y agoView on Hugging Face
1likes
app.py39 linesDownload Raw Back to root
1import streamlit as st2from transformers import pipeline3 4# Load the translation pipeline with the specified model5pipe = pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")6 7# Define language codes8language_codes = {9    "Bengali": "bn_IN",10    "Gujarati": "gu_IN",11    "Kannada": "kn_IN",12    "Telugu": "te_IN",13    "English": "en_XX"14}15 16# Streamlit app17st.title("Multilingual Translation App")18 19# Input text20input_text = st.text_area("Enter text to translate:")21 22# Language selection23target_language = st.selectbox("Select target language:", options=list(language_codes.keys()))24 25# Translation button26if st.button("Translate"):27    if input_text:28        # Get the target language code29        tgt_lang = language_codes[target_language]30        31        # Perform translation32        translated_text = pipe(input_text, src_lang="en_XX", tgt_lang=tgt_lang)[0]['translation_text']33        34        # Display translated text35        st.subheader("Translated Text:")36        st.write(translated_text)37    else:38        st.write("Please enter text to translate.")39