vamsibanda
commited on
Commit
•
2fd65d8
1
Parent(s):
1cc9751
Update README.md
Browse files
README.md
CHANGED
@@ -27,4 +27,5 @@ model = ORTModelForCustomTasks.from_pretrained("vamsibanda/sbert-all-MiniLM-L12-
|
|
27 |
tokenizer = AutoTokenizer.from_pretrained("vamsibanda/sbert-all-MiniLM-L12-with-pooler")
|
28 |
inputs = tokenizer("I love burritos!", return_tensors="pt")
|
29 |
pred = model(**inputs)
|
|
|
30 |
```
|
|
|
27 |
tokenizer = AutoTokenizer.from_pretrained("vamsibanda/sbert-all-MiniLM-L12-with-pooler")
|
28 |
inputs = tokenizer("I love burritos!", return_tensors="pt")
|
29 |
pred = model(**inputs)
|
30 |
+
embedding = pred['pooler_output']
|
31 |
```
|