Update README.md

README.md

@@ -29,7 +29,7 @@ from depth_anything.dpt import DepthAnything
 from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from torchvision.transforms import Compose
 
-model = DepthAnything.from_pretrained("LiheYoung/depth_anything_vits14")
+model = DepthAnything.from_pretrained("LiheYoung/depth_anything_vitl14")
 
 transform = Compose([
         Resize(
@@ -45,8 +45,9 @@ transform = Compose([
         PrepareForNet(),
     ])
 
-img = Image.open(...)
-image = transform({'image': np.array(image)})['image']
+image = Image.open("...")
+image = np.array(image) / 255.0
+image = transform({'image': image})['image']
 image = torch.from_numpy(image).unsqueeze(0)
 
 depth = model(image)