SiraH commited on
Commit
0cdaf3d
1 Parent(s): e968908

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -90,12 +90,12 @@ def main():
90
  if "messages" not in st.session_state:
91
  st.session_state.messages = []
92
 
93
- repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
94
- llm = HuggingFaceHub(
95
- repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_length": 128})
96
 
97
 
98
- # llm = load_llama2_llamaCpp()
99
  qa_prompt = set_custom_prompt()
100
  embeddings = load_embeddings()
101
 
 
90
  if "messages" not in st.session_state:
91
  st.session_state.messages = []
92
 
93
+ # repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
94
+ # llm = HuggingFaceHub(
95
+ # repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_length": 128})
96
 
97
 
98
+ llm = load_llama2_llamaCpp()
99
  qa_prompt = set_custom_prompt()
100
  embeddings = load_embeddings()
101