oracat commited on
Commit
e908864
·
1 Parent(s): 8dcbe0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def process(text):
45
  """
46
  Translate incoming text to tokens and classify it
47
  """
48
- pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
49
  result = pipe(text)[0]
50
  return format_predictions(top_pct(result))
51
 
 
45
  """
46
  Translate incoming text to tokens and classify it
47
  """
48
+ pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None)
49
  result = pipe(text)[0]
50
  return format_predictions(top_pct(result))
51