beta

app.py

@@ -43,7 +43,7 @@ def main_pipeline(image):
     #image = Image.open(image)
 
     #image = image.convert("RGB")
-
+    original_image = image
     image = np.asarray(image)
 
     gray_img = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
@@ -83,15 +83,12 @@ def main_pipeline(image):
     data = {}
     data['age'] = str(age_prds[0][0])
     data['gender'] = gender
-    b_box_arr = [(x, y, w, h) for (x, y, w, h) in faces]
 
-    data['left'] = str(b_box_arr[0][0])
-    data['top'] =  str(b_box_arr[0][1])
-    data['width'] = str(b_box_arr[0][2])
-    data['height'] = str(b_box_arr[0][3])
-    predictions_json = json.dumps(data)
+    image_b_box = cv2.rectangle(np.asarray(original_image), (x, y), (x+w, y+h), (255, 0, 0), 2)
 
-    return predictions_json
+    result_image = Image.fromarray(image_b_box)
+    result_image = result_image.resize((150,150))
+    return result_image, data['age'], data['gender'] 
 
 # initializing the input component
 image = gr.inputs.Image(shape = (224, 224))
@@ -101,4 +98,6 @@ labels = gr.outputs.Label()
  
 # launching the interface
 gr.Interface(fn = main_pipeline, inputs = image,
-             outputs = labels, capture_session = True).launch()
+             outputs = ['image',
+                        'text', 
+                        'text'], capture_session = True).launch()