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','пример текста']))
|