Update periodontitis_detection.py

periodontitis_detection.py

@@ -316,7 +316,7 @@ class SimpleDentalSegmentationNoEnhance:
 
 if __name__ == "__main__":
     unet_model = "models/unet/best.keras"  # Keras model
-    yolo_model = "models/yolov8/best.pt"               # YOLOv8 PyTorch model
+    yolo_model = "best2.pt"               # YOLOv8 PyTorch model
     image_path = "trial.jpg"
 
     seg = SimpleDentalSegmentationNoEnhance(unet_model, yolo_model)