Update app.py

app.py

@@ -191,6 +191,13 @@ def resize_dimensions(dimensions, target_size):
 def flush():
     gc.collect()
     torch.cuda.empty_cache()
+
+@spaces.GPU
+def clear_gpu():
+    pipe = pipe.to("cpu")
+    guide_pipe = guide_pipe.to("cpu")
+    depth_estimator = depth_estimator.to("cpu")
+    torch.cuda.empty_cache()
     
 class ControlNetDepthDesignModelMulti:
     """ Produces random noise images """
@@ -332,12 +339,6 @@ 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')
-
-    def clear_gpu():
-        pipe = pipe.to("cpu")
-        guide_pipe = guide_pipe.to("cpu")
-        depth_estimator = depth_estimator.to("cpu")
-        torch.cuda.empty_cache()
         
     def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, 
                   img_size, detr_confidence_threshold=0.8):