Spaces:
Sleeping
Sleeping
Synced repo using 'sync_with_huggingface' Github Action
Browse files
var.py
CHANGED
@@ -51,7 +51,7 @@ class SchemaVectorDB:
|
|
51 |
class FullVectorDB:
|
52 |
def __init__(self):
|
53 |
self.client = chromadb.PersistentClient(path="db/chroma")
|
54 |
-
|
55 |
|
56 |
# Get existing collections or create if not exist
|
57 |
self.restaurants_col = self.client.get_or_create_collection("restaurants")
|
|
|
51 |
class FullVectorDB:
|
52 |
def __init__(self):
|
53 |
self.client = chromadb.PersistentClient(path="db/chroma")
|
54 |
+
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="./my_cache")
|
55 |
|
56 |
# Get existing collections or create if not exist
|
57 |
self.restaurants_col = self.client.get_or_create_collection("restaurants")
|