Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -27,8 +27,7 @@ model = model.to(device)
27
 
28
  def single_sentence(sentence):
29
  predictions = predict_single(sentence, tokenizer, model, device)
30
- predictions.sort(reverse=True)
31
- return list(zip(LABEL_COLUMNS, predictions))
32
 
33
  def csv_process(csv_file, attr="content"):
34
  current_time = datetime.now()
 
27
 
28
  def single_sentence(sentence):
29
  predictions = predict_single(sentence, tokenizer, model, device)
30
+ return sorted(zip(LABEL_COLUMNS, predictions), key=lambda x:x[-1], reverse=True)
 
31
 
32
  def csv_process(csv_file, attr="content"):
33
  current_time = datetime.now()