add examples

app.py

@@ -26,7 +26,10 @@ parts_dict = {
     "corgi": "head,leg,body,ear",
     "pushcar": "wheel,body,handle",
     "plant": "pot,plant",
-    "chair": "back of chair,leg,seat"
+    "chair": "back of chair,leg,seat",
+    "objpart_redblack": "head,arm,foot,body,leg,hand,knapsack,neck",
+    "objpart_dragon": "body,head,leg,wing,tail,foot",
+    "objpart_catgirl": "Body,Leg,Head,Hair,Foot,Hand,Ear,Shorts,Tail,Arm"
 }
 
 source_dict = {
@@ -40,7 +43,10 @@ source_dict = {
     "corgi": "wild",
     "pushcar": "wild",
     "plant": "wild",
-    "chair": "wild"
+    "chair": "wild",
+    "objpart_redblack": "wild",
+    "objpart_catgirl": "wild",
+    "objpart_dragon": "wild"
 }
 
 @spaces.GPU(duration=120)
@@ -118,7 +124,7 @@ with gr.Blocks(theme=gr.themes.Default(text_size="lg", radius_size="none")) as d
             output_point_cloud = gr.Plot(label="Output Result")
 
     with gr.Row(variant="panel"):
-        with gr.Row(scale=6):
+        with gr.Column(scale=6):
             title = gr.HTML('''<h1 text-align="center">Objaverse</h1>
         <p style='font-size: 16px;'>Online 3D assets from Objaverse!</p>
         ''')
@@ -128,18 +134,21 @@ with gr.Blocks(theme=gr.themes.Default(text_size="lg", radius_size="none")) as d
                                   ("examples/objaverse/motorvehicle.jpg", "motor vehicle"),
                                   ("examples/objaverse/teddy.jpg", "teddy bear"),
                                   ("examples/objaverse/shirt.jpg", "shirt")],
-                                  columns=6,
+                                  columns=3,
                                   allow_preview=False)
             gallery_objaverse.select(fn=on_select, 
                            inputs=None, 
                            outputs=[file_upload, part_queries])
-        with gr.Row(scale=6):
+        with gr.Column(scale=6):
             title = gr.HTML("""<h1 text-align="center">In the Wild</h1>
-        <p style='font-size: 16px;'>Challenging in-the-wild reconstructions from iPhone photos!</p>
+        <p style='font-size: 16px;'>Challenging examples in-the-wild reconstructions from iPhone photos and PartObjaverseTiny!</p>
         """)
             gallery_wild = gr.Gallery([("examples/wild/pushcar.jpg", "iPhone-pushcar"),
-                                  ("examples/wild/plant.jpg", "iPhone-plant")],
-                                  columns=6,
+                                  ("examples/wild/plant.jpg", "iPhone-plant"),
+                                  ("examples/wild/objpart_catgirl.jpg", "objpart_catgirl"),
+                                  ("examples/wild/objpart_dragon.jpg", "objpart_dragon"),
+                                  ("examples/wild/objpart_redblack.jpg", "objpart_redblack"),],
+                                  columns=3,
                                   allow_preview=False)
             gallery_wild.select(fn=on_select, 
                            inputs=None,