Update gradio_demo.py

gradio_demo.py

@@ -784,7 +784,9 @@ class MaskMover:
         result.paste(fg, (x, y), fg)  # Use fg as the mask (requires fg to be in 'RGBA' mode)
         
         return np.array(result.convert('RGB'))  # Convert back to 'RGB' if needed
-        
+
+@spaces.GPU(duration=60)
+@torch.inference_mode()
 def get_depth(image):
     if image is None:
         return None