vedsadani commited on
Commit
09b10d4
·
verified ·
1 Parent(s): b89e20f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,11 +21,12 @@ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-
21
  vector_store= FAISS.load_local("vector_db/", embeddings)
22
 
23
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
24
- llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.01, "max_new_tokens": 4096})
25
 
26
  retriever = vector_store.as_retriever(
27
  search_type="similarity",
28
- search_kwargs={"k":5, "include_metadata": True})
 
29
 
30
  df=pd.read_csv('data/Gretel_Data.csv')
31
  averages = df.mean().to_dict()
 
21
  vector_store= FAISS.load_local("vector_db/", embeddings)
22
 
23
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
24
+ llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.01, "max_new_tokens": 2048})
25
 
26
  retriever = vector_store.as_retriever(
27
  search_type="similarity",
28
+ search_kwargs={"k":3, "include_metadata": True}
29
+ )
30
 
31
  df=pd.read_csv('data/Gretel_Data.csv')
32
  averages = df.mean().to_dict()