geetu040 commited on
Commit
c1c2261
·
1 Parent(s): 78cf6f6

update checkpoint

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