Update README.md

README.md

@@ -11,9 +11,12 @@ tags:
 inference: false
 ---
     
-# SDXL-controlnet: Depth
+# SDXL-controlnet: Zoe-Depth
 
-These are controlnet weights trained on stabilityai/stable-diffusion-xl-base-1.0 with depth conditioning. You can find some example images in the following. 
+These are ControlNet weights trained on stabilityai/stable-diffusion-xl-base-1.0 with zoe depth conditioning. [Zoe-depth](https://github.com/isl-org/ZoeDepth) is an open-source  
+SOTA depth estimation model which produces high-quality depth maps, which are better suited for conditioning.
+
+You can find some example images in the following. 
 
 ![images_0)](./zoe-depth-example.png)
 
@@ -82,7 +85,7 @@ def colorize(value, vmin=None, vmax=None, cmap='gray_r', invalid_val=-99, invali
 def get_zoe_depth_map(image):
     with torch.autocast("cuda", enabled=True):
         depth = model_zoe_n.infer_pil(image)
-    depth = colorize(depth, cmap="gray_r", gamma_corrected=True)
+    depth = colorize(depth, cmap="gray_r")
     return depth
 ```