Update app.py

app.py

@@ -210,7 +210,7 @@ class ControlNetDepthDesignModelMulti:
         self.neg_prompt = "window, door, low resolution, banner, logo, watermark, text, deformed, blurry, out of focus, surreal, ugly, beginner"
         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:
         """
@@ -339,7 +339,8 @@ def create_demo(model):
             design_image = gr.Image(label="Output Mask", elem_id='img-display-output')
         with gr.Column():
             bboxes = gr.JSON(label="Detected Objects", elem_id='json-display-output')
-        
+
+    @spaces.GPU
     def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, 
                   img_size, detr_confidence_threshold=0.8):