Update app.py

app.py

@@ -1,15 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the model
 MODEL_AGE = pipeline('image-classification', model='nateraw/vit-age-classifier', device=-1)
 
 def predict(image):
     age_result = MODEL_AGE(image)
-    # Get the top prediction
     top_age = age_result[0]['label']
-    # You can return a number or a string, but your iOS app expects a number
-    # Let's map the label to a representative age (e.g., the midpoint of the range)
     age_map = {
         "3-9": 6,
         "10-19": 15,