Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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=
|
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=
|
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,
|