Update README.md

README.md

@@ -159,19 +159,13 @@ blurred_image = gaussian_blur(img)
 #### Depth
 
 ```python
-# install depthfm from https://github.com/CompVis/depth-fm
-import torchvision.transforms as transforms
-from depthfm.dfm import DepthFM
-depthfm_model = DepthFM(ckpt_path=checkpoint_path)
-depthfm_model.eval()
+# install image_gen_aux with: pip install git+https://github.com/asomoza/image_gen_aux.git
+from image_gen_aux import DepthPreprocessor
 
-# assuming img is a PIL image
-img = F.to_tensor(img)
-c, h, w = img.shape
-img = F.interpolate(img, (512, 512), mode='bilinear', align_corners=False)
-with torch.no_grad():
-  img = self.depthfm_model(img, num_steps=2, ensemble_size=4)
-img = F.interpolate(img, (h, w), mode='bilinear', align_corners=False)
+image = load_image("path to image")
+
+depth_preprocessor = DepthPreprocessor.from_pretrained("depth-anything/Depth-Anything-V2-Large-hf").to("cuda")
+depth_image = depth_preprocessor(image, invert=True)[0].convert("RGB")
 ```
 
 ### Tips