MatteoScript commited on
Commit
2587659
1 Parent(s): 57ffc82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -21,7 +21,7 @@ import tqdm
21
  import accelerate
22
 
23
 
24
- # default_persist_directory = './chroma_HF/'
25
 
26
  llm_name0 = "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
 
@@ -54,7 +54,7 @@ def create_db(splits, collection_name):
54
  embedding=embedding,
55
  client=new_client,
56
  collection_name=collection_name,
57
- # persist_directory=default_persist_directory
58
  )
59
  return vectordb
60
 
@@ -74,9 +74,6 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
74
 
75
  # HuggingFaceHub uses HF inference endpoints
76
  progress(0.5, desc="Initializing HF Hub...")
77
- # Use of trust_remote_code as model_kwargs
78
- # Warning: langchain issue
79
- # URL: https://github.com/langchain-ai/langchain/issues/6080
80
  if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.1":
81
  llm = HuggingFaceHub(
82
  repo_id=llm_model,
 
21
  import accelerate
22
 
23
 
24
+ default_persist_directory = './chromaDB/'
25
 
26
  llm_name0 = "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
 
 
54
  embedding=embedding,
55
  client=new_client,
56
  collection_name=collection_name,
57
+ persist_directory=default_persist_directory
58
  )
59
  return vectordb
60
 
 
74
 
75
  # HuggingFaceHub uses HF inference endpoints
76
  progress(0.5, desc="Initializing HF Hub...")
 
 
 
77
  if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.1":
78
  llm = HuggingFaceHub(
79
  repo_id=llm_model,