Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,12 +18,13 @@ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-
|
|
18 |
vector_store= FAISS.load_local("vector_db/", embeddings)
|
19 |
|
20 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
|
21 |
-
llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.
|
22 |
|
23 |
memory = ConversationSummaryBufferMemory(
|
24 |
llm=llm,
|
25 |
output_key='answer',
|
26 |
memory_key='chat_history',
|
|
|
27 |
return_messages=True)
|
28 |
|
29 |
retriever = vector_store.as_retriever(
|
|
|
18 |
vector_store= FAISS.load_local("vector_db/", embeddings)
|
19 |
|
20 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
|
21 |
+
llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 4096})
|
22 |
|
23 |
memory = ConversationSummaryBufferMemory(
|
24 |
llm=llm,
|
25 |
output_key='answer',
|
26 |
memory_key='chat_history',
|
27 |
+
max_token_limit=100,
|
28 |
return_messages=True)
|
29 |
|
30 |
retriever = vector_store.as_retriever(
|