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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -26,11 +26,11 @@ def load_clip_model():
26
  clip_model, preprocess_val, tokenizer, device = load_clip_model()
27
 
28
  # Load YOLOv8 model
29
- @st.cache_resource
30
- def load_yolo_model():
31
- return YOLO("./best.pt")
32
 
33
- yolo_model = load_yolo_model()
34
 
35
  # Load chromaDB
36
  client = chromadb.PersistentClient(path="./clothesDB_202410")
 
26
  clip_model, preprocess_val, tokenizer, device = load_clip_model()
27
 
28
  # Load YOLOv8 model
29
+ #@st.cache_resource
30
+ #def load_yolo_model():
31
+ # return YOLO("./best.pt")
32
 
33
+ #yolo_model = load_yolo_model()
34
 
35
  # Load chromaDB
36
  client = chromadb.PersistentClient(path="./clothesDB_202410")