Add application file

app.py

@@ -435,14 +435,14 @@ def main():
             """)
     
     # Load or train model
-    # if 'model' not in st.session_state:
-    #     with st.spinner("Loading AI model... Please wait..."):
-    #         model, tokenizer = load_or_train_model()
-    #         if model is None or tokenizer is None:
-    #             st.error("Failed to load/train the AI model. Please try again.")
-    #             return
-    #         st.session_state.model = model
-    #         st.session_state.tokenizer = tokenizer
+    if 'model' not in st.session_state:
+        with st.spinner("Loading AI model... Please wait..."):
+            model, tokenizer = load_or_train_model()
+            if model is None or tokenizer is None:
+                st.error("Failed to load/train the AI model. Please try again.")
+                return
+            st.session_state.model = model
+            st.session_state.tokenizer = tokenizer
     
     # Create two columns for input form
     col1, col2 = st.columns([2, 1])