Update app.py

app.py

@@ -175,26 +175,7 @@ if st.session_state.framework == "gen":
     main_bg = "picturebg.png"
 
 
-    if st.session_state.framework == "lstm":
-        bg_color = "#FF5733"  # For example, a warm red/orange
-        bg_color_iv = "orange"  # For example, a warm red/orange
-        text_h1 = "BI-DIRECTIONAL"
-        text_i = "Long short term memory"
-        model = "TENSORFLOW"
-        st.session_state.model = "best_bilstm_model.h5"
-    if st.session_state.framework == "gru":
-        bg_color = "#FF5733"  # For example, a warm red/orange
-        bg_color_iv = "orange"  # For example, a warm red/orange
-        text_h1 = "GATED RECURRENT UNIT"
-        text_i = "Recurrent Neural Network"
-        model = "TENSORFLOW"
-        st.session_state.model = "best_gru_model.h5"
-    if st.session_state.framework == "gen":
-        bg_color = "#FF5733"  # For example, a warm red/orange
-        bg_color_iv = "orange"  # For example, a warm red/orange
-        text_h1 = "Amazon Stock Predictor"
-        text_i = "21 Days Ahead of the Market"
-        model = "TENSORFLOW"
+  
     st.markdown(
         f"""
         <style>
@@ -1499,6 +1480,7 @@ if st.session_state.framework == "dash":
                             if st.button(f"🗂 {truncated_preview}", key=val):
                                 load_chat_history(chat_id)
                                 st.session_state.show_overlay = False
+                                st.rerun()
 
 
                         with col2: