Update README.md

README.md

@@ -28,16 +28,16 @@ You can find some example images in the following.
 Make sure first to install the libraries:
 
 ```bash
-pip install accelerate transformers safetensors diffusers
+pip install accelerate transformers safetensors diffusers timm==0.6.7
 ```
 
 And then setup the zoe-depth model
 
 ```
 import torch
-import matplotlib
 import matplotlib.cm
 import numpy as np
+from PIL import Image
 
 torch.hub.help("intel-isl/MiDaS", "DPT_BEiT_L_384", force_reload=True)  # Triggers fresh download of MiDaS repo
 model_zoe_n = torch.hub.load("isl-org/ZoeDepth", "ZoeD_NK", pretrained=True).eval()
@@ -96,9 +96,6 @@ Now we're ready to go:
 
 ```python
 import torch
-import numpy as np
-from PIL import Image
-
 from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
 from diffusers.utils import load_image