Update app.py color

app.py

@@ -123,9 +123,17 @@ def builder(lang, text):
             logit = model(input_ids=tokenized_word['input_ids'], 
                 attention_mask=tokenized_word['attention_mask']).logits
         word_output = m(logit)
-        if word_output[0][1] > 0.95:
+        if word_output[0][1] > 0.99:
+            output_analysis.append( (word, '+++') )
+        elif word_output[0][1] > 0.9:
+            output_analysis.append( (word, '++') )
+        elif word_output[0][1] > 0.8:
             output_analysis.append( (word, '+') )
-        elif word_output[0][1] < 0.05:
+        elif word_output[0][1] < 0.01:
+            output_analysis.append( (word, '---') )
+        elif word_output[0][1] < 0.1:
+            output_analysis.append( (word, '--') )
+        elif word_output[0][1] < 0.2:
             output_analysis.append( (word, '-') )
         else:
             output_analysis.append( (word, None) )
@@ -148,7 +156,8 @@ def builder(lang, text):
 demo = gr.Interface(builder, inputs=[gr.inputs.Dropdown(['Any', 'Eng', 'Kor']), "text"], 
                     outputs=[ gr.Label(num_top_classes=3, label='Lang'), 
                             gr.Label(num_top_classes=2, label='Result'),
-                            gr.HighlightedText(label="Analysis", combine_adjacent=False).style(color_map={"+": "red", "-": "green"}) ],
+                            gr.HighlightedText(label="Analysis", combine_adjacent=False)
+                            .style(color_map={"+++": "#CF0000", "++": "#FF3232", "+": "#FFD4D4", "---": "#0600CF", "--": "#4C47FF", "-": "#BEBDFF"}) ],
                     # outputs='label',
                     title=title, description=description, examples=examples)