Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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":
|
25 |
|
26 |
retriever = vector_store.as_retriever(
|
27 |
search_type="similarity",
|
28 |
-
search_kwargs={"k":
|
|
|
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()
|