Spaces:
Sleeping
Sleeping
Changed model path
Browse files
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
|