Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
|