ritampatra/Translator_PROO
1
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 