Update app.py

app.py

@@ -47,10 +47,13 @@ def pose_estimation(test_image):
 
     canvas = util.draw_handpose(canvas, all_hand_peaks)
 
-    plt.imshow(canvas[:, :, [2, 1, 0]])
-    plt.axis('off')
+    # plt.imshow(canvas[:, :, [2, 1, 0]])
+    # plt.axis('off')
+    # out_image_path = './out.jpg'
+    # plt.savefig(out_image_path)
+
     out_image_path = './out.jpg'
-    plt.savefig(out_image_path)
+    cv2.imwrite(out_image_path, canvas) 
 
     # Save JSON data and return its path
     json_file_path = save_json(candidate, subset)