Update app.py

app.py

@@ -7,6 +7,15 @@ import pickle
 
 st.write(f"Using TensorFlow version: {tf.__version__}")
 
+@st.cache_resource
+def load_translation_model():
+    try:
+        return load_model("translation_model.h5")
+    except Exception as e:
+        st.error(f"Error loading model: {e}")
+        return None
+
+@st.cache_resource
 def load_tokenizers():
     with open("en_tokenizer.pkl", "rb") as f:
         en_tokenizer = pickle.load(f)
@@ -23,24 +32,23 @@ def translate_text(text, model, en_tokenizer, fr_tokenizer, en_max_len):
 
 st.title("English to French Translator")
 st.write("Enter an English sentence, and the model will generate the French translation.")
-try:
-    model = load_model("translation_model.h5")
-except:
-    pass
 
+model = load_translation_model()
 en_tokenizer, fr_tokenizer = load_tokenizers()
 en_max_len = 55
 
-if "default_text" not in st.session_state:
-    st.session_state.default_text = ""
+if "user_input" not in st.session_state:
+    st.session_state.user_input = ""
 
 if st.button("Use Example Sentence"):
-    st.session_state.default_text = "this world is bad"
+    st.session_state.user_input = "This world is bad"
 
-user_input = st.text_input("Enter an English sentence:", st.session_state.default_text)
+user_input = st.text_input("Enter an English sentence:", value=st.session_state.user_input)
 
 if st.button("Translate"):
-    if user_input:
+    if model is None:
+        st.write("Error: Model not loaded.")
+    elif user_input:
         translation = translate_text(user_input, model, en_tokenizer, fr_tokenizer, en_max_len)
         st.write(f"**French Translation:** {translation}")
     else: