sanpdy commited on
Commit
15ab5d3
·
1 Parent(s): 9bdbc7e

Changed model path

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -70,7 +70,8 @@ print(f"Using device: {device}")
70
  # Load YOLO detector
71
  #yolo = YOLO("/home/sankalp/flake_classification/models/best.pt")
72
  #yolo = YOLO("/home/sankalp/yolo_flake_detection/yolo11n_synthetic_runs/exp1/weights/best.pt")
73
- yolo = YOLO("/home/sankalp/yolo_flake_detection/yolo_runs/yolo11l_flake_runs/weights/best.pt")
 
74
  yolo.conf = 0.5
75
 
76
  # Load classifier weights
 
70
  # Load YOLO detector
71
  #yolo = YOLO("/home/sankalp/flake_classification/models/best.pt")
72
  #yolo = YOLO("/home/sankalp/yolo_flake_detection/yolo11n_synthetic_runs/exp1/weights/best.pt")
73
+ #yolo = YOLO("/home/sankalp/yolo_flake_detection/yolo_runs/yolo11l_flake_runs/weights/best.pt")
74
+ yolo = YOLO("models/yolo-flake-detector.pt")
75
  yolo.conf = 0.5
76
 
77
  # Load classifier weights