yslan commited on
Commit
5b1668b
·
1 Parent(s): c305f12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -126,7 +126,7 @@ pipeline = get_pipeline(model_id, "unet", model_config['down_block_types'], mode
126
  # pipeline = None
127
 
128
  ### run the demo ##
129
- @spaces.GPU(duration=50)
130
  def segment(canvas, image, logits):
131
  if logits is not None:
132
  logits *= 32.0
@@ -167,7 +167,7 @@ def segment(canvas, image, logits):
167
 
168
  return mask[0], masked_img, masked_img, logits / 32.0
169
 
170
- @spaces.GPU(duration=50)
171
  def get_depth(image, points):
172
 
173
  depth = d_model_NK.infer_pil(image)
@@ -186,7 +186,7 @@ def get_depth(image, points):
186
  return depth, depth_img, colored_depth[:, :, :3]
187
 
188
 
189
- @spaces.GPU(duration=50)
190
  def run_objctrl_2_5d(condition_image,
191
  mask,
192
  depth,
 
126
  # pipeline = None
127
 
128
  ### run the demo ##
129
+ @spaces.GPU(duration=5)
130
  def segment(canvas, image, logits):
131
  if logits is not None:
132
  logits *= 32.0
 
167
 
168
  return mask[0], masked_img, masked_img, logits / 32.0
169
 
170
+ @spaces.GPU(duration=5)
171
  def get_depth(image, points):
172
 
173
  depth = d_model_NK.infer_pil(image)
 
186
  return depth, depth_img, colored_depth[:, :, :3]
187
 
188
 
189
+ @spaces.GPU(duration=80)
190
  def run_objctrl_2_5d(condition_image,
191
  mask,
192
  depth,