Update app.py

app.py

@@ -9,8 +9,10 @@ app.prepare(ctx_id=0, det_size=(640, 640))
 def calculate(photo):
     image = cv2.imread(photo)
     faces = app.get(image)
+    image_draw = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     faceid_embeds = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
-    return faceid_embeds
+    image_draw = app.draw_on(image_draw, faces)
+    return image_draw
     
 with gr.Blocks() as demo:
     with gr.Row():
@@ -18,9 +20,10 @@ with gr.Blocks() as demo:
             face_photo = gr.Image(label="Photo", type="filepath")
             greet_btn = gr.Button("Calculate")
         with gr.Column():
+            output_image = gr.Image(label="Output")
             output = gr.JSON()
     
-    greet_btn.click(fn=calculate, inputs=face_photo, outputs=output, api_name="calculate_face_embedding")
+    greet_btn.click(fn=calculate, inputs=face_photo, outputs=output_image, api_name="calculate_face_embedding")
 
 if __name__ == "__main__":
     demo.launch()