jayn95 commited on
Commit
2942407
·
verified ·
1 Parent(s): 8b81f1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from periodontitis_detection import SimpleDentalSegmentationNoEnhance
8
  # ==========================
9
  model = SimpleDentalSegmentationNoEnhance(
10
  unet_model_path="unet.keras", # same filenames as your repo
11
- yolo_model_path="best.pt"
12
  )
13
 
14
  # ==========================
 
8
  # ==========================
9
  model = SimpleDentalSegmentationNoEnhance(
10
  unet_model_path="unet.keras", # same filenames as your repo
11
+ yolo_model_path="best2.pt"
12
  )
13
 
14
  # ==========================