Spaces:

JovianSP
/

OIS-AI-Defect

Sleeping

jovian commited on Nov 13, 2024

Commit

372a23a

1 Parent(s): b856007

bes.pt

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ class Detection:
     def __init__(self):
         # Set the model path and confidence threshold
-        yolov8_model_path = "./model/company_model.pt"  # Update to your model path
         # Initialize the AutoDetectionModel
         self.model = AutoDetectionModel.from_pretrained(

     def __init__(self):
         # Set the model path and confidence threshold
+        yolov8_model_path = "./model/best.pt"  # Update to your model path
         # Initialize the AutoDetectionModel
         self.model = AutoDetectionModel.from_pretrained(