Update app.py

app.py

@@ -1,13 +1,10 @@
 import streamlit as st
 from transformers import pipeline
 
-# Set up the translator using a Hugging Face model (you can replace this with the model of your choice)
-@st.cache(allow_output_mutation=True)
-def load_model():
-    return pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr")
-
-# Load the model
-translator = load_model()
+# Set up the translator using a Hugging Face model
+@st.cache_resource
+def load_model(model_name):
+    return pipeline("translation", model=model_name)
 
 # Title of the app
 st.title("Language Translator App")
@@ -32,7 +29,7 @@ target_language = st.selectbox("Target Language", list(language_options.keys()))
 if st.button("Translate"):
     # Load the chosen model based on the selected language
     selected_model = language_options[target_language]
-    translator = pipeline("translation", model=selected_model)
+    translator = load_model(selected_model)
     
     # Perform translation
     if input_text:
@@ -44,3 +41,4 @@ if st.button("Translate"):
         st.write(translated_text)
     else:
         st.write("Please enter some text to translate.")
+