Update periodontitis_detection.py

periodontitis_detection.py

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