Spaces:
Sleeping
Sleeping
jovian commited on
Commit ·
9b6d8c1
1
Parent(s): 19ca7c4
change model
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ class Detection:
|
|
| 15 |
|
| 16 |
def __init__(self):
|
| 17 |
# Set the model path and confidence threshold
|
| 18 |
-
yolov8_model_path = "./model/
|
| 19 |
|
| 20 |
# Initialize the AutoDetectionModel
|
| 21 |
self.model = AutoDetectionModel.from_pretrained(
|
|
|
|
| 15 |
|
| 16 |
def __init__(self):
|
| 17 |
# Set the model path and confidence threshold
|
| 18 |
+
yolov8_model_path = "./model/best_100epochs_latest.pt" # Update to your model path
|
| 19 |
|
| 20 |
# Initialize the AutoDetectionModel
|
| 21 |
self.model = AutoDetectionModel.from_pretrained(
|