Create app.py

app.py

@@ -0,0 +1,43 @@
+import streamlit as st
+from transformers import pipeline
+
+# Caching the model to avoid reloading it multiple times
+@st.cache_resource
+def load_model():
+    return pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")
+
+# Load the translation pipeline
+pipe = load_model()
+
+# Define language codes
+language_codes = {
+    "Bengali": "bn_IN",
+    "Gujarati": "gu_IN",
+    "Kannada": "kn_IN",
+    "Telugu": "te_IN",
+    "English": "en_XX"
+}
+
+# Streamlit app
+st.title("Multilingual Translation App")
+
+# Input text
+input_text = st.text_area("Enter text to translate:")
+
+# Language selection
+target_language = st.selectbox("Select target language:", options=list(language_codes.keys()))
+
+# Translation button
+if st.button("Translate"):
+    if input_text:
+        # Get the target language code
+        tgt_lang = language_codes[target_language]
+        
+        # Perform translation
+        translated_text = pipe(input_text, src_lang="en_XX", tgt_lang=tgt_lang)[0]['translation_text']
+        
+        # Display translated text
+        st.subheader("Translated Text:")
+        st.write(translated_text)
+    else:
+        st.write("Please enter text to translate.")