Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -87,7 +87,7 @@ def ade_palette() -> list[list[int]]:
|
|
| 87 |
[102, 255, 0], [92, 0, 255]]
|
| 88 |
|
| 89 |
|
| 90 |
-
@torch.inference_mode
|
| 91 |
@spaces.GPU
|
| 92 |
def get_depth_image(image: Image) -> Image:
|
| 93 |
|
|
@@ -115,7 +115,7 @@ def get_depth_image(image: Image) -> Image:
|
|
| 115 |
image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
|
| 116 |
return image
|
| 117 |
|
| 118 |
-
@torch.inference_mode
|
| 119 |
@spaces.GPU
|
| 120 |
def get_segmentation_of_room(image: Image):
|
| 121 |
#-> tuple[np.ndarray, Image]:
|
|
@@ -144,7 +144,7 @@ def get_segmentation_of_room(image: Image):
|
|
| 144 |
seg_image = Image.fromarray(color_seg).convert('RGB')
|
| 145 |
return seg_image
|
| 146 |
|
| 147 |
-
@torch.inference_mode
|
| 148 |
@spaces.GPU
|
| 149 |
def interior_inference(image,
|
| 150 |
prompt,
|
|
|
|
| 87 |
[102, 255, 0], [92, 0, 255]]
|
| 88 |
|
| 89 |
|
| 90 |
+
@torch.inference_mode()
|
| 91 |
@spaces.GPU
|
| 92 |
def get_depth_image(image: Image) -> Image:
|
| 93 |
|
|
|
|
| 115 |
image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
|
| 116 |
return image
|
| 117 |
|
| 118 |
+
@torch.inference_mode()
|
| 119 |
@spaces.GPU
|
| 120 |
def get_segmentation_of_room(image: Image):
|
| 121 |
#-> tuple[np.ndarray, Image]:
|
|
|
|
| 144 |
seg_image = Image.fromarray(color_seg).convert('RGB')
|
| 145 |
return seg_image
|
| 146 |
|
| 147 |
+
@torch.inference_mode()
|
| 148 |
@spaces.GPU
|
| 149 |
def interior_inference(image,
|
| 150 |
prompt,
|