JoJosmin commited on
Commit
4c2dfc1
1 Parent(s): 3053989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,9 +33,9 @@ def load_yolo_model():
33
  yolo_model = load_yolo_model()
34
 
35
  # Load chromaDB
36
- client = chromadb.PersistentClient(path="./clothesDB_Test")
37
  #collection = client.get_collection(name="clothes_items_ver3")
38
- collection = client.get_collection(name="category_upper")
39
 
40
 
41
  # Helper functions
 
33
  yolo_model = load_yolo_model()
34
 
35
  # Load chromaDB
36
+ client = chromadb.PersistentClient(path="./clothesDB_202410")
37
  #collection = client.get_collection(name="clothes_items_ver3")
38
+ collection = client.get_collection(name="clothes")
39
 
40
 
41
  # Helper functions