Spaces:

poltextlab
/

babel_machine

Running

poltextlab commited on May 6

Commit

2490733

verified ·

1 Parent(s): 471013f

fix dict

Files changed (1) hide show

interfaces/cap_minor.py CHANGED Viewed

@@ -84,7 +84,7 @@ def predict(text, model_id, tokenizer_id):
     html = ""
     first = True
-    for label, prob in first_n_items.items():
         bar_color = "#e0d890" if first else "#ccc"
         text_color = "black"
         bar_width = int(prob * 100)

     html = ""
     first = True
+    for label, prob in first_n_items:
         bar_color = "#e0d890" if first else "#ccc"
         text_color = "black"
         bar_width = int(prob * 100)