focusit commited on
Commit
cec0fe3
1 Parent(s): 89cb144

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ docs = splitter.split_documents(documents)
20
  embeddings = HuggingFaceEmbeddings()
21
  doc_search = Chroma.from_documents(docs, embeddings)
22
 
23
- repo_id = "tiiuae/falcon-7b"
24
  llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 4000})
25
 
26
  from langchain.schema import retriever
 
20
  embeddings = HuggingFaceEmbeddings()
21
  doc_search = Chroma.from_documents(docs, embeddings)
22
 
23
+ repo_id = "mistralai/Mistral-7B-v0.1"
24
  llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 4000})
25
 
26
  from langchain.schema import retriever