apple
/

DepthPro-mixin

Depth Estimation

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions Community

amael-apple commited on 12 days ago

Commit

943bb94

•

1 Parent(s): 41f26b3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -80,7 +80,7 @@ class DepthProWrapper(DepthPro, PyTorchModelHubMixin):
 # Load model and preprocessing transform
-model = DepthProWrapper.from_pretrained("DepthPro-L")
 transform = Compose(
         [
             ToTensor(),

 # Load model and preprocessing transform
+model = DepthProWrapper.from_pretrained("apple/DepthPro-mixin")
 transform = Compose(
         [
             ToTensor(),