Update app.py

app.py

@@ -15,8 +15,9 @@ from diffusers import StableDiffusionXLPipeline
 import gradio as gr
 import gc
 
-device = "cuda"
-dtype = torch.float16
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
         
 css = """
@@ -75,7 +76,6 @@ def get_segmentation_pipeline(
 
 
 @torch.inference_mode()
-@spaces.GPU
 def segment_image(
         image: Image,
         image_processor: AutoImageProcessor,
@@ -120,7 +120,6 @@ def get_depth_pipeline():
 
 
 @torch.inference_mode()
-@spaces.GPU
 def get_depth_image(
         image: Image,
         feature_extractor: AutoImageProcessor,
@@ -188,7 +187,6 @@ class ControlNetDepthDesignModelMulti:
         self.control_items = ["windowpane;window", "door;double;door"]
         self.additional_quality_suffix = "interior design, 4K, high resolution, photorealistic"
         
-    @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