add white bg

interfaces/cap_minor.py

@@ -88,6 +88,7 @@ def predict(text, model_id, tokenizer_id):
     first_n_items = take(5, output_pred.items())
 
     html = ""
+    html += '<div style="background-color: white">'
     first = True
     for label, prob in first_n_items:
         bar_color = "#e0d890" if first else "#ccc"
@@ -96,6 +97,8 @@ def predict(text, model_id, tokenizer_id):
 
         bar_color = score_to_color(prob)
 
+        
+
         if first:
             html += f"""
             <div style="text-align: center; font-weight: bold; font-size: 30px; margin-bottom: 10px;">
@@ -109,8 +112,9 @@ def predict(text, model_id, tokenizer_id):
         </div>
         """
         first = False
+        
     
-    
+    html += '</div>'
     output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
     return html, output_info