amael-apple
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ class DepthProWrapper(DepthPro, PyTorchModelHubMixin):
|
|
80 |
|
81 |
|
82 |
# Load model and preprocessing transform
|
83 |
-
model = DepthProWrapper.from_pretrained("DepthPro-
|
84 |
transform = Compose(
|
85 |
[
|
86 |
ToTensor(),
|
|
|
80 |
|
81 |
|
82 |
# Load model and preprocessing transform
|
83 |
+
model = DepthProWrapper.from_pretrained("apple/DepthPro-mixin")
|
84 |
transform = Compose(
|
85 |
[
|
86 |
ToTensor(),
|