Spaces:
Runtime error
Runtime error
fix age detector
Browse files
main.py
CHANGED
|
@@ -16,7 +16,7 @@ PY_MODULES = {
|
|
| 16 |
"skin_tone.py": "SkinToneDetector",
|
| 17 |
"oiliness.py": "OilinessDetector",
|
| 18 |
"wrinkle_unet.py": "WrinkleDetector",
|
| 19 |
-
"age.py": "
|
| 20 |
}
|
| 21 |
|
| 22 |
def load_model(token):
|
|
@@ -97,7 +97,7 @@ def analyze_skin(image: np.ndarray, analysis_type: str) -> np.ndarray:
|
|
| 97 |
output = detector.draw_json(result)
|
| 98 |
|
| 99 |
elif analysis_type == "Skin Age":
|
| 100 |
-
detector = detector_classes["
|
| 101 |
result = detector.predict_json()
|
| 102 |
output = detector.draw_json(result)
|
| 103 |
|
|
|
|
| 16 |
"skin_tone.py": "SkinToneDetector",
|
| 17 |
"oiliness.py": "OilinessDetector",
|
| 18 |
"wrinkle_unet.py": "WrinkleDetector",
|
| 19 |
+
"age.py": "AgePredictor"
|
| 20 |
}
|
| 21 |
|
| 22 |
def load_model(token):
|
|
|
|
| 97 |
output = detector.draw_json(result)
|
| 98 |
|
| 99 |
elif analysis_type == "Skin Age":
|
| 100 |
+
detector = detector_classes["AgePredictor"](image)
|
| 101 |
result = detector.predict_json()
|
| 102 |
output = detector.draw_json(result)
|
| 103 |
|