Daniel Ferreira commited on
Commit
327bec4
1 Parent(s): 25503ec

use top_k instead of return_all_scores

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ pipe = opt_pipeline(
24
  function_to_apply="sigmoid",
25
  accelerator="ort",
26
  tokenizer=tokenizer,
27
- return_all_scores=True, # 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','пример текста']))
 
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','пример текста']))