depth map black

app.py

@@ -31,12 +31,12 @@ async def predict_depth(file: UploadFile = File(...)):
         print("[DEBUG] Realizando predicción de profundidad...")
         depth = pipe(image)
 
-        # Visualiza la profundidad
-        vis = pipe.image_processor.visualize_depth(depth.prediction)
+        # Exporta la profundidad como una imagen 16-bit PNG
+        depth_16bit = pipe.image_processor.export_depth_to_16bit_png(depth.prediction)
 
         # Guarda la imagen generada en un buffer
         img_buffer = io.BytesIO()
-        vis[0].save(img_buffer, format="PNG")
+        depth_16bit[0].save(img_buffer, format="PNG")
         img_buffer.seek(0)
 
         # Devuelve la imagen como respuesta