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(),
|