broadened color range

app.py

@@ -17,15 +17,17 @@ if __name__ == "__main__":
 
     # Define your color-coding labels; if prob > x, then label = y; Sorted in descending probability order!
     probs_to_label = [
-        (0.1, "p >= 10%"),
-        (0.01, "p >= 1%"),
-        (1e-20, "p < 1%"),
+        (0.8, "p >= 80%"),
+        (0.5, "p >= 50%"),
+        (0.2, "p >= 20%"),
+        (0.0, "p >= 0%")
     ]
 
     label_to_color = {
-        "p >= 10%": "green",
-        "p >= 1%": "yellow",
-        "p < 1%": "red"
+        "p >= 80%": "blue",
+        "p >= 50%": "green",
+        "p >= 20%": "yellow",
+        "p >= 0%": "red"
     }
 
     def get_tokens_and_labels(prompt):
@@ -103,4 +105,4 @@ if __name__ == "__main__":
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()