patrickvonplaten commited on
Commit
347e7ac
·
1 Parent(s): 118967a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -83,10 +83,12 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCM
83
  from PIL import Image
84
  import numpy as np
85
  import torch
 
86
 
87
  depth_estimator = pipeline('depth-estimation')
88
 
89
- image = Image.open('./images/stormtrooper.png')
 
90
  image = depth_estimator(image)['depth']
91
  image = np.array(image)
92
  image = image[:, :, None]
 
83
  from PIL import Image
84
  import numpy as np
85
  import torch
86
+ from diffusers.utils import load_image
87
 
88
  depth_estimator = pipeline('depth-estimation')
89
 
90
+ image = load_image("https://huggingface.co/lllyasviel/sd-controlnet-depth/resolve/main/images/stormtrooper.png")
91
+
92
  image = depth_estimator(image)['depth']
93
  image = np.array(image)
94
  image = image[:, :, None]