Lam-Hung commited on
Commit
095bf58
1 Parent(s): 0173054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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,