update bar

interfaces/cap_minor.py

@@ -17,6 +17,11 @@ def take(n, iterable):
     """Return the first n items of the iterable as a list."""
     return list(islice(iterable, n))
 
+def score_to_color(prob):
+    red = int(255 * (1 - prob))
+    green = int(255 * prob)
+    return f"rgb({red},{green},0)"
+
 HF_TOKEN = os.environ["hf_read"]
 
 languages = [
@@ -89,6 +94,8 @@ def predict(text, model_id, tokenizer_id):
         text_color = "black"
         bar_width = int(prob * 100)
 
+        bar_color = score_to_color(prob)
+
         if first:
             html += f"""
             <div style="display: flex; bold; font-size: 30px; justify-content: space-between; align-items: center; margin-bottom: 4px;">
@@ -101,7 +108,7 @@ def predict(text, model_id, tokenizer_id):
             <span style="color: {text_color};">{label}</span>
             <span style="color: {text_color};">{int(prob * 100)}%</span>
         </div>
-        <div style="height: 4px; background: linear-gradient(to right, red, green); width: {bar_width}%; margin-bottom: 8px;"></div>
+        <div style="height: 4px; background-color: {bar_color}; width: {bar_width}%; margin-bottom: 8px;"></div>
         """
         first = False