nielsr HF staff commited on
Commit
cc7a571
·
verified ·
1 Parent(s): 06e1d62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -59,8 +59,8 @@ import requests
59
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
60
  image = Image.open(requests.get(url, stream=True).raw)
61
 
62
- image_processor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-small-hf")
63
- model = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-small-hf")
64
 
65
  # prepare image for the model
66
  inputs = image_processor(images=image, return_tensors="pt")
 
59
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
60
  image = Image.open(requests.get(url, stream=True).raw)
61
 
62
+ image_processor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-base-hf")
63
+ model = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-base-hf")
64
 
65
  # prepare image for the model
66
  inputs = image_processor(images=image, return_tensors="pt")