adarsh
commited on
Commit
·
4eef5e8
1
Parent(s):
b91a6f2
model path fixed
Browse files
app.py
CHANGED
|
@@ -57,7 +57,7 @@ class TreeDetectionModel:
|
|
| 57 |
|
| 58 |
|
| 59 |
# Load the trained YOLOv8 model
|
| 60 |
-
model_path = "./yolov8_model/
|
| 61 |
tree_detector = TreeDetectionModel(model_path)
|
| 62 |
|
| 63 |
def detect_trees(image):
|
|
|
|
| 57 |
|
| 58 |
|
| 59 |
# Load the trained YOLOv8 model
|
| 60 |
+
model_path = "./yolov8_model/best_model.pt"
|
| 61 |
tree_detector = TreeDetectionModel(model_path)
|
| 62 |
|
| 63 |
def detect_trees(image):
|