Spaces:

leedoming
/

itda-nosegmentation

Runtime error

leedoming commited on Sep 10

Commit

97480db

•

1 Parent(s): c4d252d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ clip_model, preprocess_val, tokenizer, device = load_clip_model()
 # Load YOLOv8 model
 @st.cache_resource
 def load_yolo_model():
-    return YOLO("./yolov8m.pt")
 yolo_model = load_yolo_model()

 # Load YOLOv8 model
 @st.cache_resource
 def load_yolo_model():
+    return YOLO("./best.pt")
 yolo_model = load_yolo_model()