Update app.py

app.py

@@ -74,6 +74,7 @@ def predict(image, model, face_detector, device, margin=0.4, input_size=224):
     # detect faces using dlib detector
     detected = face_detector(image, 1)
     faces = np.empty((len(detected), input_size, input_size, 3))
+    age_data = []
 
     if len(detected) > 0:
         for i, d in enumerate(detected):
@@ -98,8 +99,9 @@ def predict(image, model, face_detector, device, margin=0.4, input_size=224):
 
         # draw results
         for age, d in zip(predicted_ages, detected):
-            draw_label(image, (d.left(), d.top()), f'{int(age)}')
-    return image
+            age_text = f'{int(age)}'
+            age_data.append({'age': int(age), 'text': age_text, 'face_coordinates': (d.left(), d.top())})
+    return age_data
 
 
 def main():