Spaces:
Runtime error
Runtime error
update
Browse files
model.py
CHANGED
|
@@ -631,7 +631,8 @@ class Model:
|
|
| 631 |
127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
|
| 632 |
|
| 633 |
results = [x_samples[i] for i in range(num_samples)]
|
| 634 |
-
|
|
|
|
| 635 |
|
| 636 |
@torch.inference_mode()
|
| 637 |
def process_depth(self, input_image, prompt, a_prompt, n_prompt,
|
|
|
|
| 631 |
127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
|
| 632 |
|
| 633 |
results = [x_samples[i] for i in range(num_samples)]
|
| 634 |
+
|
| 635 |
+
return [detected_map] + results
|
| 636 |
|
| 637 |
@torch.inference_mode()
|
| 638 |
def process_depth(self, input_image, prompt, a_prompt, n_prompt,
|