change scoring logic

app.py

@@ -20,18 +20,14 @@ def iterate_sentences(x):
 
 
 def racism_analysis(text):
-    results = racism_analysis_pipe(text)[0]
-    label = "Non-racist" if results["label"] == "LABEL_0" else "Racist"
-    score = (
-        1 - results["score"]
-        if results["label"] == "LABEL_0"
-        else results["score"]
-    )
+    results = racism_analysis_pipe(text, return_all_scores=True)[0][1]
+    label = "Non-racist" if results["score"] < 0.5 else "Racist"
+    score = results["score"]
 
     words = {}
     colors = []
     for word, sentence in iterate_sentences(text):
-        words[word] = 1 / \
+        words[word] = score / \
             racism_analysis_pipe(sentence, return_all_scores=True)[
             0][1]["score"]
         color = "+" if words[word] > 100 else None