beta beautify

app.py

@@ -81,14 +81,15 @@ def main_pipeline(image):
         gender = 'female'
 
     data = {}
-    data['age'] = str(age_prds[0][0])
+    data['age'] = int(age_prds[0][0])
     data['gender'] = gender
 
     image_b_box = cv2.rectangle(np.asarray(original_image), (x, y), (x+w, y+h), (255, 0, 0), 2)
 
     result_image = Image.fromarray(image_b_box)
     result_image = result_image.resize((150,150))
-    return result_image, data['age'], data['gender'] 
+
+    return result_image, f"~ {(data['age'])}", data['gender'] 
 
 # initializing the input component
 image = gr.inputs.Image(shape = (224, 224))
@@ -97,7 +98,7 @@ labels = gr.outputs.Label()
 
  
 # launching the interface
-gr.Interface(fn = main_pipeline, inputs = image,
-             outputs = ['image',
-                        'text', 
-                        'text'], capture_session = True).launch()
+gr.Interface(fn = main_pipeline, title= 'Age & Gender predictions', inputs = image,
+             outputs = [gr.Image(label= 'Result'),
+                        gr.Text(label= 'Age'), 
+                        gr.Text(label= 'Gender')], flagging_options=["correct", "incorrect", "other"], capture_session = True).launch()