jovian commited on
Commit
9b6d8c1
·
1 Parent(s): 19ca7c4

change model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/best.pt" # Update to your model path
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(