Update README.md (#5)
Browse files- Update README.md (a6875881d41f3442904fc8372537878cc2023386)
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-
|
32 |
-
model = DPTForDepthEstimation.from_pretrained("Intel/dpt-
|
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")
|