IFMedTechdemo commited on
Commit
a8f7f15
·
verified ·
1 Parent(s): 18399f8

fix age detector

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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": "AgeDetector"
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["AgeDetector"](image)
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