spaces

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import gradio as gr
 

objctrl_2_5d/objctrl_2_5d.py

@@ -1,3 +1,5 @@
+import spaces
+
 import os
 import torch
 import numpy as np
@@ -39,6 +41,7 @@ down_scale = 8
 H, W = height // down_scale, width // down_scale
 K = np.array([[width / down_scale, 0, W / 2], [0, width / down_scale, H / 2], [0, 0, 1]])
 
+@spaces.GPU(duration=50)
 def run(pipeline, device):
     def run_objctrl_2_5d(condition_image, 
                          mask, 

objctrl_2_5d/utils/ui_utils.py

@@ -1,3 +1,5 @@
+import spaces
+
 import gradio as gr
 from PIL import Image
 import numpy as np
@@ -50,6 +52,7 @@ def process_image(raw_image):
 def get_subject_points(canvas):
     return canvas["image"], canvas["points"]
 
+@spaces.GPU(duration=50)
 def run_segment(segmentor):
     def segment(canvas, image, logits):
         if logits is not None:
@@ -142,6 +145,7 @@ def get_points(img,
 def undo_points(original_image):
     return original_image, []
 
+@spaces.GPU(duration=50)
 def run_depth(d_model_NK):
     def get_depth(image, points):