Update app.py

app.py

@@ -2,8 +2,9 @@ import os
 os.system('python setup.py develop')
 
 import argparse
-from pathlib import Path
 import json
+from pathlib import Path
+
 import gradio as gr
 import matplotlib
 
@@ -139,23 +140,18 @@ with gr.Blocks() as demo:
              })
              ],
             ['examples/sofa1.jpg',
-             'examples/sofa2.jpg',
-             json.dumps({
-                 'points': [(112, 328), (120, 260), (104, 208), (200, 260), (208, 336), (216, 212), (208, 172),
-                            (284, 180), (276, 324), (156, 308), (172, 228), (256, 232), (168, 184), (260, 196)]
-
-,
-                 'skeleton': [(0, 1), (3, 1), (3, 4), (10, 9), (11, 8),
-                              (1, 10), (10, 11), (11, 3), (1, 2), (7, 6),
-                              (5, 13), (5, 3), (13, 11), (12, 10), (12, 2),
-                              (6, 10), (7, 11)],
+             'examples/sofa2.png',
+             json.dumps({'points': [[272, 561], [193, 482], [339, 460], [445, 530], [264, 369], [203, 318], [354, 300],
+                                    [457, 341], [345, 63], [187, 68]],
+                         'skeleton': [[0, 4], [1, 5], [2, 6], [3, 7], [7, 6], [6, 5],
+                                      [5, 4], [4, 7], [5, 9], [9, 8], [8, 6]],
              })],
             ['examples/person1.jpeg',
              'examples/person2.jpeg',
              json.dumps({
-                 'points': [(380, 484), (640, 488), (520, 616), (424, 736), (612, 724)],
-                 'skeleton': [(0, 1), (1, 2), (0, 2), (2, 3), (2, 4),
-                              (4, 3)],
+                 'points': [[322, 488], [431, 486], [526, 644], [593, 486], [697, 492], [407, 728],
+                            [522, 726], [625, 737], [515, 798]],
+                 'skeleton': [[0, 1], [1, 3], [3, 4], [1, 2], [2, 3], [5, 6], [6, 7], [7, 8], [8, 5]],
              })]
         ],
             inputs=[support_image, query_image, example_null],
@@ -192,4 +188,3 @@ if __name__ == "__main__":
     print("Start app", parser.parse_args())
     gr.close_all()
     demo.launch(show_api=False)
-