vamsibanda
commited on
Commit
•
1cc9751
1
Parent(s):
1f889db
Update README.md
Browse files
README.md
CHANGED
@@ -27,12 +27,4 @@ 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 |
-
```
|
31 |
-
You will also be able to leverage the pipeline API in transformers:
|
32 |
-
```python
|
33 |
-
from transformers import pipeline
|
34 |
-
|
35 |
-
onnx_extractor = pipeline("feature-extraction", model=model, tokenizer=tokenizer)
|
36 |
-
text = "I love burritos!"
|
37 |
-
pred = onnx_extractor(text)
|
38 |
```
|
|
|
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 |
```
|