Update app.py

app.py

@@ -81,9 +81,7 @@ with st.sidebar:
                     docs = loader.load_data()
 
                     # setup llm & embedding model
-                    llm = st.session_state.llm  # Load the Cerebras model from session state
-                    if not llm:
-                        st.stop()  # Stop execution if model initialization failed
+                    llm = load_llm()
                     embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-large-en-v1.5", trust_remote_code=True)
                     # Creating an index over loaded data
                     Settings.embed_model = embed_model