Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
| 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 |
# ====================================================
|