init

app.py

@@ -13,6 +13,7 @@ from utils import map_colors_rgb
 from diffusers import StableDiffusionXLPipeline
 import gradio as gr
 import gc
+import spaces
 
 device = "cuda"
 dtype = torch.float16
@@ -209,7 +210,8 @@ class ControlNetDepthDesignModelMulti:
         self.seg_image_processor, self.image_segmentor = get_segmentation_pipeline()
         self.depth_feature_extractor, self.depth_estimator = get_depth_pipeline()
         self.depth_estimator = self.depth_estimator.to(device)
-
+    
+    @spaces.GPU
     def generate_design(self, empty_room_image: Image, prompt: str, guidance_scale: int = 10, num_steps: int = 50, strength: float =0.9, img_size: int = 640) -> Image:
         """
         Given an image of an empty room and a prompt