Tom Aarsen commited on
Commit ·
4fb4c43
1
Parent(s): e08714e
Use transformers load_image as the repository's load_image doesn't have user-agent
Browse files
README.md
CHANGED
|
@@ -135,6 +135,7 @@ This model supports **text**, **image**, and **image+text** embedding with Sente
|
|
| 135 |
|
| 136 |
```python
|
| 137 |
from sentence_transformers import SentenceTransformer
|
|
|
|
| 138 |
|
| 139 |
model = SentenceTransformer("nvidia/llama-nemotron-embed-vl-1b-v2", trust_remote_code=True)
|
| 140 |
|
|
@@ -146,9 +147,9 @@ documents = [
|
|
| 146 |
]
|
| 147 |
# Images can be URLs, file paths, or PIL Images
|
| 148 |
images = [
|
| 149 |
-
"https://developer.download.nvidia.com/images/isaac/nvidia-isaac-lab-1920x1080.jpg",
|
| 150 |
-
"https://blogs.nvidia.com/wp-content/uploads/2018/01/automotive-key-visual-corp-blog-level4-av-og-1280x680-1.png",
|
| 151 |
-
"https://developer-blogs.nvidia.com/wp-content/uploads/2025/02/hc-press-evo2-nim-25-featured-b.jpg",
|
| 152 |
]
|
| 153 |
|
| 154 |
# Text-only encoding
|
|
|
|
| 135 |
|
| 136 |
```python
|
| 137 |
from sentence_transformers import SentenceTransformer
|
| 138 |
+
from transformers.image_utils import load_image
|
| 139 |
|
| 140 |
model = SentenceTransformer("nvidia/llama-nemotron-embed-vl-1b-v2", trust_remote_code=True)
|
| 141 |
|
|
|
|
| 147 |
]
|
| 148 |
# Images can be URLs, file paths, or PIL Images
|
| 149 |
images = [
|
| 150 |
+
load_image("https://developer.download.nvidia.com/images/isaac/nvidia-isaac-lab-1920x1080.jpg"),
|
| 151 |
+
load_image("https://blogs.nvidia.com/wp-content/uploads/2018/01/automotive-key-visual-corp-blog-level4-av-og-1280x680-1.png"),
|
| 152 |
+
load_image("https://developer-blogs.nvidia.com/wp-content/uploads/2025/02/hc-press-evo2-nim-25-featured-b.jpg"),
|
| 153 |
]
|
| 154 |
|
| 155 |
# Text-only encoding
|