Update app.py

app.py

@@ -1,10 +1,15 @@
 import streamlit as st
 from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 
-# Load the model and tokenizer
-model_name = "facebook/m2m100_418M"
-tokenizer = M2M100Tokenizer.from_pretrained(model_name)
-model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+# Load the model and tokenizer once at the start
+@st.cache_resource(allow_output_mutation=True)
+def load_model_and_tokenizer():
+    model_name = "facebook/m2m100_418M"
+    tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+    model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+    return model, tokenizer
+
+model, tokenizer = load_model_and_tokenizer()
 
 # Streamlit UI
 st.title("English to Multiple Language Translator")
@@ -13,7 +18,10 @@ st.write("Translate English text into different languages using AI.")
 # Input text
 input_text = st.text_area("Enter English text:", value="")
 
-# Language selection
+# Sidebar for language selection
+st.sidebar.title("Settings")
+st.sidebar.subheader("Select Target Language")
+
 language_options = {
     "French": "fr",
     "Spanish": "es",
@@ -26,7 +34,8 @@ language_options = {
     "Portuguese": "pt",
     "Italian": "it"
 }
-selected_language = st.selectbox("Select target language:", list(language_options.keys()))
+
+selected_language = st.sidebar.selectbox("Target Language", list(language_options.keys()))
 
 if st.button("Translate"):
     if input_text:
@@ -46,3 +55,4 @@ if st.button("Translate"):
         st.write(translated_text)
     else:
         st.write("Please enter text to translate.")
+