Update app.py

app.py

@@ -11,6 +11,7 @@ import imageio
 import requests
 import argparse
 import imageio
+import spaces
 from scipy.spatial.transform import Rotation  
 
 from gradio_draggable import Draggable
@@ -105,7 +106,8 @@ def objs_to_canvas(lst, length=256, scale = 2.6):
         canvas[mask] = feat
     canvas = np.transpose(canvas, [2, 0, 1]).astype(np.float32)
     return canvas
- 
+
+@spaces.GPU
 @torch.no_grad()
 def predict_local_view(lst):
     canvas = torch.tensor(objs_to_canvas(lst)).cuda()[None]
@@ -115,6 +117,7 @@ def predict_local_view(lst):
     rgb = gen['gen_output']['image'][0] * .5 + .5
     return to_pil(rgb)
 
+@spaces.GPU
 @torch.no_grad()
 def predict_local_view_video(lst):
     canvas = torch.tensor(objs_to_canvas(lst)).cuda()[None]
@@ -140,6 +143,7 @@ def predict_local_view_video(lst):
     writer.close()
     return 'tmp.mp4'
 
+@spaces.GPU
 @torch.no_grad()
 def predict_global_view(lst):
     canvas = torch.tensor(objs_to_canvas(lst)).cuda()[None]