Update app.py

app.py

@@ -192,12 +192,12 @@ 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()
+# @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 """
@@ -286,6 +286,11 @@ class ControlNetDepthDesignModelMulti:
             (orig_w, orig_h), Image.Resampling.LANCZOS
         )
         
+        pipe = pipe.to("cpu")
+        guide_pipe = guide_pipe.to("cpu")
+        depth_estimator = depth_estimator.to("cpu")
+        torch.cuda.empty_cache()
+        
         return design_image