lmao numpy array

app.py

@@ -93,10 +93,12 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
     
     output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, depth_image=depth_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
     
-    output_depth_vis = (output.depth[0] - np.min(output.depth[0])) / (np.max(output.depth[0]) - np.min(output.depth[0])) * 255
+    depth_out = np.array(output.depth[0])
+    
+    output_depth_vis = (depth_out - np.min(output.depth[0])) / (np.max(depth_out) - np.min(depth_out)) * 255
     output_depth_vis = output_depth_vis.astype("uint8")
     output_depth = Image.fromarray(output_depth_vis)
-    
+
     return output.rgb[0], output_depth, gr.update(visible=True)