Update app.py

app.py

@@ -81,7 +81,7 @@ 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
+    return image, str(predicted_ages)
 
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -100,6 +100,7 @@ with gr.Blocks(css="style.css") as demo:
             run_button = gr.Button("Run")
         with gr.Column():
             result = gr.Image(label="Result")
+            digits = gr.Textbox()
 
     gr.Examples(
         examples=examples,