alexxxey123 commited on
Commit
ec02655
·
verified ·
1 Parent(s): 665b792

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ def classify_email(text):
107
  for model_name in model_names:
108
  response += f"{model_name}: "
109
  for j, prob in enumerate(probabilities[model_name]):
110
- response += f"{num_to_label[j]}: {prob:.2f}%"
111
  response += "\n"
112
 
113
  return response
 
107
  for model_name in model_names:
108
  response += f"{model_name}: "
109
  for j, prob in enumerate(probabilities[model_name]):
110
+ response += f"{labels[j]}: {prob:.2f}%"
111
  response += "\n"
112
 
113
  return response