Update app.py

app.py

@@ -4,7 +4,6 @@ import trimesh
 import traceback
 import numpy as np
 import gradio as gr
-from functools import partial
 from multiprocessing import Process, Queue
 
 import torch
@@ -144,12 +143,12 @@ def py_string_to_mesh_file_safe(py_string, mesh_path):
         target=py_string_to_mesh_file,
         args=(py_string, mesh_path, queue))
     process.start()
-    process.join(5)
+    process.join(7)
 
     if process.is_alive():
         process.terminate()
         process.join()
-        raise gr.Error('Process is alive after 3 seconds')
+        raise gr.Error('Process is alive after 7 seconds')
     
     if not queue.empty():
         raise gr.Error(queue.get())
@@ -203,9 +202,20 @@ def run_mesh(py_string):
 def run():
     with gr.Blocks() as demo:
         with gr.Row():
-            gr.Markdown('## CAD-Recode Demo\n'
-                        'Upload mesh or select from examples and press Run! Mesh ⇾ 256 points ⇾ Python code by CAD-Recode ⇾ CAD model.')
+            with gr.Column(scale=2):
+                gr.Markdown(
+                    '## CAD-Recode Demo\n'
+                    'Upload mesh or select from examples and press Run! Mesh ⇾ 256 points ⇾ Python code by CAD-Recode ⇾ CAD model.')
 
+            with gr.Column(scale=1):
+                with gr.Row():
+                    gr.Image('./logos/snt.png', height=30, min_width=90, show_label=False, show_download_button=False, show_fullscreen_button=False, show_share_button=False)
+                    gr.Image('./logos/fnr.png', height=30, min_width=90, show_label=False, show_download_button=False, show_fullscreen_button=False, show_share_button=False)
+                    gr.Image('./logos/a3d.png', height=30, min_width=90, show_label=False, show_download_button=False, show_fullscreen_button=False, show_share_button=False)
+                
+                with gr.Row():
+                    gr.Markdown('GitHub: [filaPro/cad-recode](https://github.com/filaPro/cad-recode), arXiv: [2412.14042](https://arxiv.org/abs/2412.14042)')
+        
         with gr.Row(equal_height=True):
             in_model = gr.Model3D(label='1. Input Mesh', interactive=True)
             point_model = gr.Model3D(label='2. Sampled Point Cloud', display_mode='point_cloud', interactive=False)
@@ -227,12 +237,23 @@ def run():
                             ['./data/21492_8bd34fc1_0008.stl', 42],
                             ['./data/00375556.stl', 96],
                             ['./data/49121_adb01620_0000.stl', 42],
-                            ['./data/41473_c2137170_0023.stl', 42]],
+                            ['./data/41473_c2137170_0023.stl', 42],
+                            ['./data/22447_4062c6cb_0011.stl', 42],
+                            ['./data/27694_7801dc67_0017.stl', 45],
+                            ['./data/49562_6df35938_0005.stl', 44],
+                            ['./data/131709_8b86dfb6_0000.stl', 45],
+                            ['./data/00081523.stl', 44],
+                            ['./data/00614972.stl', 44],
+                            ['./data/User Library-Cople Ventosa V2.stl', 50],
+                            ['./data/User Library-Caster, Superior Brand, 600lb 2 EDIT.stl', 52]],
                         example_labels=[
                             'fusion360_table1', 'deepcad_star', 'cc3d_gear', 'deepcad_barrels',
-                            'fusion360_gear', 'deepcad_house', 'fusion360_table2', 'fusion360_omega'],
+                            'fusion360_gear', 'deepcad_house', 'fusion360_table2', 'fusion360_omega',
+                            'fusion360_hat', 'fusion360_table3', 'fusion360_bolt', 'fusion360_clamp',
+                            'deepcad_alpha', 'deepcad_pulley', 'cc3d_flange', 'cc3d_caster'],
                         inputs=[in_model, seed_slider],
-                        cache_examples=False)
+                        cache_examples=False,
+                        examples_per_page=20)
                 with gr.Row():
                     run_button = gr.Button('Run')
 
@@ -270,4 +291,4 @@ cad_recode = CADRecode.from_pretrained(
     attn_implementation='flash_attention_2').eval()
 
 os.environ['TOKENIZERS_PARALLELISM'] = 'False'
-run()
+run()