Daniel Ferreira
commited on
Commit
·
327bec4
1
Parent(s):
25503ec
use top_k instead of return_all_scores
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ pipe = opt_pipeline(
|
|
| 24 |
function_to_apply="sigmoid",
|
| 25 |
accelerator="ort",
|
| 26 |
tokenizer=tokenizer,
|
| 27 |
-
|
| 28 |
)
|
| 29 |
|
| 30 |
print(pipe(['example text','exemple de texte','texto de ejemplo','testo di esempio','texto de exemplo','örnek metin','пример текста']))
|
|
|
|
| 24 |
function_to_apply="sigmoid",
|
| 25 |
accelerator="ort",
|
| 26 |
tokenizer=tokenizer,
|
| 27 |
+
top_k=None, # return scores for all the labels, model was trained as multilabel
|
| 28 |
)
|
| 29 |
|
| 30 |
print(pipe(['example text','exemple de texte','texto de ejemplo','testo di esempio','texto de exemplo','örnek metin','пример текста']))
|