try extra function again

app.py

@@ -5,8 +5,14 @@ from __future__ import annotations
 import os
 
 import gradio as gr
+import spaces
 
-from model import Model
+
+@spaces.GPU
+def run_on_gpu(input_shape, text, seed, guidance_scale):
+    from model import Model
+    model = Model()
+    return model.run(shape_path=input_shape, text=text, seed=seed, guidance_scale=guidance_scale)
 
 
 def main():
@@ -16,9 +22,11 @@ def main():
     - Inference takes about 10 minutes on a T4 GPU.
     '''
     if (SPACE_ID := os.getenv('SPACE_ID')) is not None:
-        DESCRIPTION += f'\n<p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. <a href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>'
-
-    model = Model()
+        DESCRIPTION += (f'\n<p>For faster inference without waiting in queue, '
+                        f'you may duplicate the space and upgrade to GPU in settings. '
+                        f'<a href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true">'
+                        f'<img style="display: inline; margin-top: 0em; margin-bottom: 0em" '
+                        f'src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>')
 
     with gr.Blocks(css='style.css') as demo:
         gr.Markdown(DESCRIPTION)
@@ -67,7 +75,7 @@ def main():
                         ],
                         cache_examples=False)
 
-        run_button.click(fn=model.run,
+        run_button.click(fn=run_on_gpu,
                          inputs=[
                              input_shape,
                              text,

model.py

@@ -9,7 +9,6 @@ from typing import Generator, Optional
 
 import gradio as gr
 import trimesh
-import spaces
 
 sys.path.append('TEXTurePaper')
 
@@ -21,7 +20,6 @@ class Model:
     def __init__(self):
         self.max_num_faces = 100000
 
-    @spaces.GPU
     def load_config(self, shape_path: str, text: str, seed: int,
                     guidance_scale: float) -> TrainConfig:
         text += ', {} view'
@@ -52,11 +50,9 @@ class Model:
         subprocess.run(shlex.split(f'zip -r {out_path} {mesh_dir}'))
         return out_path
 
-    @spaces.GPU
     def run(
         self, shape_path: str, text: str, seed: int, guidance_scale: float
-    ) -> Generator[tuple[list[str], Optional[str], Optional[str], str], None,
-                   None]:
+    ) -> Generator[tuple[list[str], Optional[str], Optional[str], str], None, None]:
         if not shape_path.endswith('.obj'):
             raise gr.Error('The input file is not .obj file.')
         if not self.check_num_faces(shape_path):