Update app.py

app.py

@@ -154,7 +154,6 @@ def py_string_to_mesh_file_safe(py_string, mesh_path):
         raise RuntimeError(queue.get())
 
 
-@spaces.GPU(duration=20)
 def run_gpu(model, input_ids, attention_mask, point_cloud, pad_token_id):
     if torch.cuda.is_available():
         model = model.cuda()
@@ -201,6 +200,7 @@ def run_test(in_mesh_path, seed, results):
     results.append(out_mesh_path)
 
 
+@spaces.GPU(duration=20)
 def run_test_safe(in_mesh_path, seed):
     results, log = list(), str()
     try:
@@ -210,53 +210,51 @@ def run_test_safe(in_mesh_path, seed):
     return results + [None] * (3 - len(results)) + [log]
     
 
-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.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)
-            out_model = gr.Model3D(
-                label='4. Result CAD Model', interactive=False
-            )
-        
-        with gr.Row():
-            with gr.Column():
-                with gr.Row():
-                    seed_slider = gr.Slider(label='Random Seed', value=42, interactive=True)
-                with gr.Row():
-                    _ = gr.Examples(
-                        examples=[
-                            ['./data/49215_5368e45e_0000.stl', 42],
-                            ['./data/00882236.stl', 6],
-                            ['./data/User Library-engrenage.stl', 18],
-                            ['./data/00010900.stl', 42],
-                            ['./data/21492_8bd34fc1_0008.stl', 42],
-                            ['./data/00375556.stl', 96],
-                            ['./data/49121_adb01620_0000.stl', 42]],
-                        example_labels=[
-                            'fusion360_table1', 'deepcad_star', 'cc3d_gear', 'deepcad_barrels',
-                            'fusion360_gear', 'deepcad_house', 'fusion360_table2'],
-                        inputs=[in_model, seed_slider],
-                        cache_examples=False)
-                with gr.Row():
-                    run_button = gr.Button('Run')
-                    _ = gr.LoginButton()
-
-            with gr.Column():
-                out_code = gr.Code(language='python', label='3. Generated Python Code', wrap_lines=True, interactive=False)
-
-            with gr.Column():
-                log_textbox = gr.Textbox(label='Log', placeholder='Status: OK', interactive=False)
-        
-        run_button.click(
-            run_test_safe, inputs=[in_model, seed_slider], outputs=[point_model, out_code, out_model, log_textbox])
+os.environ['TOKENIZERS_PARALLELISM'] = 'False'
 
-    demo.launch()
 
+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.')
 
-os.environ['TOKENIZERS_PARALLELISM'] = 'False'
-run()
+    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)
+        out_model = gr.Model3D(
+            label='4. Result CAD Model', interactive=False
+        )
+    
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                seed_slider = gr.Slider(label='Random Seed', value=42, interactive=True)
+            with gr.Row():
+                _ = gr.Examples(
+                    examples=[
+                        ['./data/49215_5368e45e_0000.stl', 42],
+                        ['./data/00882236.stl', 6],
+                        ['./data/User Library-engrenage.stl', 18],
+                        ['./data/00010900.stl', 42],
+                        ['./data/21492_8bd34fc1_0008.stl', 42],
+                        ['./data/00375556.stl', 96],
+                        ['./data/49121_adb01620_0000.stl', 42]],
+                    example_labels=[
+                        'fusion360_table1', 'deepcad_star', 'cc3d_gear', 'deepcad_barrels',
+                        'fusion360_gear', 'deepcad_house', 'fusion360_table2'],
+                    inputs=[in_model, seed_slider],
+                    cache_examples=False)
+            with gr.Row():
+                run_button = gr.Button('Run')
+                _ = gr.LoginButton()
+
+        with gr.Column():
+            out_code = gr.Code(language='python', label='3. Generated Python Code', wrap_lines=True, interactive=False)
+
+        with gr.Column():
+            log_textbox = gr.Textbox(label='Log', placeholder='Status: OK', interactive=False)
+    
+    run_button.click(
+        run_test_safe, inputs=[in_model, seed_slider], outputs=[point_model, out_code, out_model, log_textbox])
+
+demo.launch()