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,
|