Update app.py

app.py

@@ -1,5 +1,6 @@
 # final one
 import torch
+import spaces
 import gradio as gr
 import os
 import numpy as np
@@ -271,6 +272,7 @@ with gr.Blocks() as demo:
     img_input.change(clear_model_viewer, inputs=None, outputs=model_output)
 
     # then, generate the mesh and video
+    @spaces.GPU
     generate_mesh_button.click(step_1_generate_obj, inputs=img_input, outputs=[obj_file_output, model_output])
     generate_video_button.click(step_2_generate_video, inputs=img_input, outputs=video_file_output)