add style

interfaces/cap_minor.py

@@ -89,12 +89,19 @@ def predict(text, model_id, tokenizer_id):
         text_color = "black"
         bar_width = int(prob * 100)
 
+        if first:
+            html += f"""
+            <div style="display: flex; bold; font-size: 30px; justify-content: space-between; align-items: center; margin-bottom: 4px;">
+                <span style="color: {text_color};">{label}</span>
+            </div>
+            """
+
         html += f"""
         <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 4px;">
             <span style="color: {text_color};">{label}</span>
             <span style="color: {text_color};">{int(prob * 100)}%</span>
         </div>
-        <div style="height: 4px; background-color: {bar_color}; width: {bar_width}%; margin-bottom: 8px;"></div>
+        <div style="height: 4px; background: linear-gradient(to right, red, green); width: {bar_width}%; margin-bottom: 8px;"></div>
         """
         first = False