vedsadani commited on
Commit
15da6e7
1 Parent(s): 36b919c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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.01, "max_new_tokens": 2048})
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(