Revert "can't find GPU -> try with extra function"

This reverts commit 24154be3efe17a875f012ddd5b2db658ebe2585e.

app.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import os
 
-import spaces
 import gradio as gr
 
 from model import Model
@@ -68,22 +67,19 @@ def main():
                         ],
                         cache_examples=False)
 
-        @spaces.GPU
-        def start_processing():  # extra function to use this decorator
-            run_button.click(fn=model.run,
-                             inputs=[
-                                 input_shape,
-                                 text,
-                                 seed,
-                                 guidance_scale,
-                             ],
-                             outputs=[
-                                 viewpoint_images,
-                                 result_3d_model,
-                                 output_file,
-                                 progress_text,
-                             ])
-        start_processing()
+        run_button.click(fn=model.run,
+                         inputs=[
+                             input_shape,
+                             text,
+                             seed,
+                             guidance_scale,
+                         ],
+                         outputs=[
+                             viewpoint_images,
+                             result_3d_model,
+                             output_file,
+                             progress_text,
+                         ])
 
     demo.queue(max_size=5)
     demo.launch(debug=True)

model.py

@@ -9,6 +9,7 @@ from typing import Generator, Optional
 
 import gradio as gr
 import trimesh
+import spaces
 
 sys.path.append('TEXTurePaper')
 
@@ -50,6 +51,7 @@ 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,