jayn95 commited on
Commit
2da1714
·
verified ·
1 Parent(s): c465b0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ from periodontitis_detection import SimpleDentalSegmentationNoEnhance
11
  # Load model
12
  # ==========================
13
  model = SimpleDentalSegmentationNoEnhance(
14
- unet_model_path="unet.keras",
15
- yolo_model_path="best2.pt"
16
  )
17
 
18
  # ====================================================
 
11
  # Load model
12
  # ==========================
13
  model = SimpleDentalSegmentationNoEnhance(
14
+ unet_model_path="models/unet/best.keras",
15
+ yolo_model_path="models/yolov8/best.pt"
16
  )
17
 
18
  # ====================================================