Create app.py

app.py

@@ -0,0 +1,57 @@
+import streamlit as st
+from transformers import MarianMTModel, MarianTokenizer
+
+# Function to load model and tokenizer based on selected languages
+def load_model_and_tokenizer(src_lang, tgt_lang):
+    model_name = f"Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}"
+    model = MarianMTModel.from_pretrained(model_name)
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    return model, tokenizer
+
+# Function to translate text
+def translate_text(text, model, tokenizer):
+    inputs = tokenizer(text, return_tensors="pt", padding=True)
+    translated = model.generate(**inputs)
+    return tokenizer.decode(translated[0], skip_special_tokens=True)
+
+# Streamlit app layout
+st.title("Language Translator App")
+st.write("Translate text between multiple languages using open-source models.")
+
+# Language options (expand this list based on available models)
+languages = {
+    "English": "en",
+    "French": "fr",
+    "Spanish": "es",
+    "German": "de",
+    "Italian": "it",
+    "Russian": "ru",
+    "Chinese": "zh"
+}
+
+# Language selection
+source_language = st.selectbox("Select Input Language:", options=list(languages.keys()))
+target_language = st.selectbox("Select Output Language:", options=list(languages.keys()))
+
+# Check if source and target languages are different
+if source_language == target_language:
+    st.warning("Input and output languages must be different.")
+else:
+    src_lang_code = languages[source_language]
+    tgt_lang_code = languages[target_language]
+
+    # Load model and tokenizer
+    with st.spinner("Loading translation model..."):
+        model, tokenizer = load_model_and_tokenizer(src_lang_code, tgt_lang_code)
+
+    # Input text
+    text_to_translate = st.text_area("Enter text to translate:")
+
+    if st.button("Translate"):
+        if text_to_translate.strip():
+            with st.spinner("Translating..."):
+                translated_text = translate_text(text_to_translate, model, tokenizer)
+                st.success("Translation Completed!")
+                st.text_area("Translated Text:", value=translated_text, height=200)
+        else:
+            st.error("Please enter text to translate.")