pcuenq HF staff commited on
Commit
3934990
·
verified ·
1 Parent(s): 92b586b

Update model name in snippet (#3)

Browse files

- Update model name (0e6b119b14a055b02cc3a04057e10047d195dfee)

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -73,8 +73,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
73
  url = 'https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg'
74
  image = Image.open(requests.get(url, stream=True).raw)
75
 
76
- image_processor = DepthProImageProcessorFast.from_pretrained("apple/depth-pro-hf")
77
- model = DepthProForDepthEstimation.from_pretrained("apple/depth-pro-hf").to(device)
78
 
79
  inputs = image_processor(images=image, return_tensors="pt").to(device)
80
 
 
73
  url = 'https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg'
74
  image = Image.open(requests.get(url, stream=True).raw)
75
 
76
+ image_processor = DepthProImageProcessorFast.from_pretrained("apple/DepthPro-hf")
77
+ model = DepthProForDepthEstimation.from_pretrained("apple/DepthPro-hf").to(device)
78
 
79
  inputs = image_processor(images=image, return_tensors="pt").to(device)
80