Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import torch
 import gradio as gr
@@ -19,6 +20,7 @@ def get_temp_path():
     output_path = f"data/gradio_temp/{timestamp}/"
     return output_path
 
+@spaces.GPU(duration=120)
 def scene_generate(rgb,num_coarse_views,num_mcs_views,mcs_rect_w,mcs_steps):
     # coarse
     vistadream.scene = Gaussian_Scene(cfg)