Follow ZeroGPT usage guidance to decorate GPU-required functions

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import re
+import spaces
 
 from leo.inference import inference
 
@@ -46,6 +47,7 @@ with gr.Blocks(title='LEO Demo') as demo:
     )
 
     # LEO task-to-plan inference wrapper
+    @spaces.GPU
     def leo_task_to_plan(task_description):
         task_input = {
             "task_description": task_description,
@@ -58,6 +60,7 @@ with gr.Blocks(title='LEO Demo') as demo:
         return plan
 
     # LEO ground inference wrapper
+    @spaces.GPU
     def leo_plan_to_masks(task_description, *action_steps):
         formatted_action_steps = [
             {"action": step, "target_id": "unknown", "label": "unknown"} for step in action_steps if step != ""
@@ -72,6 +75,7 @@ with gr.Blocks(title='LEO Demo') as demo:
         return [f"assets/mask/scene0050_00/scene0050_00_obj_{mask}.glb" for mask in masks] + ["assets/mask/scene0050_00/scene0050_00_obj_empty.glb"] * (STEP_COUNTS - len(masks))
 
     # LEO task-to-plan and ground inference wrapper
+    @spaces.GPU
     def leo_task_to_plan_and_masks(task_description):
         task_input = {
             "task_description": task_description,