karalif commited on
Commit
6160f25
verified
1 Parent(s): d584bcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,13 +16,13 @@ def predict(text):
16
  label = result['label']
17
  score = result['score']
18
  if label in ["Politeness"]:
19
- response += f"<span style='background-color:green; color:black;'>{label}</span>: {score:.3f}<br>"
20
  if label in ["Sentiment"]:
21
- response += f"<span style='background-color:yellow; color:black;'>{label}</span>: {score:.3f}<br>"
22
  if label in ["Formality"]:
23
- response += f"<span style='background-color:blue; color:black;'>{label}</span>: {score:.3f}<br>"
24
  if label in ["Toxicity"]:
25
- response += f"<span style='background-color:red; color:black;'>{label}</span>: {score:.3f}<br>"
26
  else:
27
  response += f"{label}: {score:.3f}<br>"
28
 
 
16
  label = result['label']
17
  score = result['score']
18
  if label in ["Politeness"]:
19
+ response += f"<span style='background-color:#b8e994; color:black;'>{label}</span>: {score:.3f}<br>" # Light Green
20
  if label in ["Sentiment"]:
21
+ response += f"<span style='background-color:#fff3cd; color:black;'>{label}</span>: {score:.3f}<br>" # Light Yellow
22
  if label in ["Formality"]:
23
+ response += f"<span style='background-color:#bee5eb; color:black;'>{label}</span>: {score:.3f}<br>" # Light Blue
24
  if label in ["Toxicity"]:
25
+ response += f"<span style='background-color:#f8d7da; color:black;'>{label}</span>: {score:.3f}<br>" # Light Red
26
  else:
27
  response += f"{label}: {score:.3f}<br>"
28