Update app.py

app.py

@@ -26,8 +26,8 @@ POSTGRE_URL = os.environ['POSTGRE_URL']
 @st.cache_resource
 def get_llm_chain():
     return custom_chain_with_history(
-        llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001), 
-        # llm=CustomLLM(repo_id="stabilityai/stablelm-2-zephyr-1_6b", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001), 
+        # llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001), 
+        llm=CustomLLM(repo_id="google/gemma-7b", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001), 
         # memory=st.session_state.memory.chat_memory,
         memory=st.session_state.memory
     )