tsrivallabh commited on
Commit
188c014
·
verified ·
1 Parent(s): c0506a3

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. var.py +1 -1
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
- self.model = SentenceTransformer('all-MiniLM-L6-v2')
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")