nick-leland commited on
Commit
3499e9c
·
verified ·
1 Parent(s): 86678c6

Changed the YOLO model to the best.onnx file.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -275,7 +275,7 @@ learn_bias = load_learner('model_bias.pkl')
275
  learn_fresh = load_learner('model_fresh.pkl')
276
 
277
  # Loads the YOLO Model
278
- model_bulge = YOLO("bulge_yolo_model.pt")
279
  # modelv8x = YOLO("yolov8x.pt")
280
  # modelv8n = YOLO("yolov8n.pt")
281
 
 
275
  learn_fresh = load_learner('model_fresh.pkl')
276
 
277
  # Loads the YOLO Model
278
+ model_bulge = YOLO("best.onnx")
279
  # modelv8x = YOLO("yolov8x.pt")
280
  # modelv8n = YOLO("yolov8n.pt")
281