yilunzhang commited on
Commit
01e6377
·
verified ·
1 Parent(s): f92b9b2

fixed depreciation

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ pipeline = pipeline(
17
 
18
  def predict(text):
19
  cleaned_text = clean_text(text)
20
- predictions = pipeline(cleaned_text, return_all_scores=True)[0]
21
  return {
22
  p["label"]: p["score"] for p in predictions
23
  }
 
17
 
18
  def predict(text):
19
  cleaned_text = clean_text(text)
20
+ predictions = pipeline(cleaned_text, top_k=None)[0]
21
  return {
22
  p["label"]: p["score"] for p in predictions
23
  }