Trying to fix predict_depth.

app.py

@@ -111,7 +111,7 @@ description = """This is an endpoint for some image operations for a Chub.ai sta
 but now also includes [BRIA](https://huggingface.co/briaai/RMBG-1.4) for background removal."""
 
 @spaces.GPU(duration=6)
-def predict_depth(image):
+def predict_depth_image(image):
     return model.infer_image(image)
 
 with gr.Blocks(css=css) as demo:
@@ -142,7 +142,7 @@ with gr.Blocks(css=css) as demo:
 
         h, w = image.shape[:2]
 
-        depth = predict_depth(image[:, :, ::-1])
+        depth = predict_depth_image(image[:, :, ::-1])
 
         raw_depth = Image.fromarray(depth.astype('uint16'))
         tmp_raw_depth = tempfile.NamedTemporaryFile(suffix='.png', delete=False)