nielsr HF staff commited on
Commit
37e4dfa
1 Parent(s): 6614d7d

Update README.md (#5)

Browse files

- Update README.md (a6875881d41f3442904fc8372537878cc2023386)

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -28,8 +28,8 @@ import requests
28
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
29
  image = Image.open(requests.get(url, stream=True).raw)
30
 
31
- processor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
32
- model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
33
 
34
  # prepare image for the model
35
  inputs = processor(images=image, return_tensors="pt")
 
28
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
29
  image = Image.open(requests.get(url, stream=True).raw)
30
 
31
+ processor = DPTImageProcessor.from_pretrained("Intel/dpt-beit-base-384")
32
+ model = DPTForDepthEstimation.from_pretrained("Intel/dpt-beit-base-384")
33
 
34
  # prepare image for the model
35
  inputs = processor(images=image, return_tensors="pt")