enable spaces

app.py

@@ -160,7 +160,7 @@ pipeline = get_pipeline(model_id, "unet", model_config['down_block_types'], mode
 # pipeline = None
 
 ### run the demo ##
-# @spaces.GPU(duration=5)
+@spaces.GPU(duration=5)
 def segment(canvas, image, logits):
     if logits is not None:
         logits *=  32.0
@@ -201,7 +201,7 @@ def segment(canvas, image, logits):
         
     return mask[0], {'image': masked_img, 'points': points}, logits / 32.0
 
-# @spaces.GPU(duration=80)
+@spaces.GPU(duration=80)
 def run_objctrl_2_5d(condition_image, 
                         mask, 
                         depth, 
@@ -488,7 +488,7 @@ def run_objctrl_2_5d(condition_image,
 
 
 # UI function
-# @spaces.GPU(duration=5)
+@spaces.GPU(duration=5)
 def process_image(raw_image, trajectory_points):
     
     image, points = raw_image['image'], raw_image['points']
@@ -544,7 +544,7 @@ def draw_points_on_image(img, points):
     
     return img
 
-# @spaces.GPU(duration=10)
+@spaces.GPU(duration=10)
 def from_examples(raw_input, raw_image_points, canvas, seg_image_points, selected_points_text, camera_option, mask_bk):
     
     selected_points = ast.literal_eval(selected_points_text)