Update README.md
#1
by
shaine21
- opened
README.md
CHANGED
|
@@ -1,9 +1,6 @@
|
|
| 1 |
---
|
| 2 |
-
license: other
|
| 3 |
license_name: nvclv1
|
| 4 |
license_link: LICENSE
|
| 5 |
-
datasets:
|
| 6 |
-
- ILSVRC/imagenet-1k
|
| 7 |
pipeline_tag: image-feature-extraction
|
| 8 |
---
|
| 9 |
|
|
@@ -104,7 +101,7 @@ model = AutoModel.from_pretrained("nvidia/MambaVision-T-1K", trust_remote_code=T
|
|
| 104 |
# eval mode for inference
|
| 105 |
model.cuda().eval()
|
| 106 |
|
| 107 |
-
#
|
| 108 |
url = 'http://images.cocodataset.org/val2017/000000020247.jpg'
|
| 109 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 110 |
input_resolution = (3, 224, 224) # MambaVision supports any input resolutions
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
license_name: nvclv1
|
| 3 |
license_link: LICENSE
|
|
|
|
|
|
|
| 4 |
pipeline_tag: image-feature-extraction
|
| 5 |
---
|
| 6 |
|
|
|
|
| 101 |
# eval mode for inference
|
| 102 |
model.cuda().eval()
|
| 103 |
|
| 104 |
+
# repare image for the model
|
| 105 |
url = 'http://images.cocodataset.org/val2017/000000020247.jpg'
|
| 106 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 107 |
input_resolution = (3, 224, 224) # MambaVision supports any input resolutions
|