Update main.py

main.py

@@ -15,7 +15,8 @@ PY_MODULES = {
     "texture.py": "TextureDetector",
     "skin_tone.py": "SkinToneDetector",
     "oiliness.py": "OilinessDetector", 
-    "wrinkle_unet.py": "WrinkleDetector"
+    "wrinkle_unet.py": "WrinkleDetector",
+    "age.py": "AgeDetector"
 }
 
 def load_model(token):
@@ -95,6 +96,11 @@ def analyze_skin(image: np.ndarray, analysis_type: str) -> np.ndarray:
         if result.get("detected") is not None:
             output = detector.draw_json(result)
 
+    elif analysis_type == "Skin Age":
+        detector = detector_classes["AgeDetector"](image)
+        result = detector.predict_json()
+        output = detector.draw_json(result)
+        
     return output
 
 
@@ -104,7 +110,7 @@ app = gr.Interface(
     inputs=[
         gr.Image(type="numpy", label="Upload your face image"),
         gr.Radio(
-            ["Dark Circles", "Redness", "Texture", "Skin Tone", "Oiliness", "Wrinkles"],  
+            ["Dark Circles", "Redness", "Texture", "Skin Tone", "Oiliness", "Wrinkles", "Skin Age"],  
             label="Select Skin Analysis Type"
         ),
     ],