rrevoid commited on
Commit
7853862
·
1 Parent(s): 5c1deee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def predict(outputs):
41
  top_cats.append(cat)
42
  top_probs.append(round(percent, 1))
43
 
44
- chart_data = pd.DataFrame([top_probs], columns=top_cats)
45
  st.area_chart(data=chart_data)
46
 
47
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
 
41
  top_cats.append(cat)
42
  top_probs.append(round(percent, 1))
43
 
44
+ chart_data = pd.DataFrame(top_probs, index=top_cats, columns=['percent'])
45
  st.area_chart(data=chart_data)
46
 
47
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")