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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,10 +39,10 @@ def predict(outputs):
39
  percent = prob * 100
40
  top += percent
41
  top_cats.append(cat)
42
- top_probs.append(prob)
43
 
44
  chart_data = pd.DataFrame([top_probs], columns=top_cats)
45
- st.bar_chart(chart_data)
46
 
47
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
48
  model = init_model()
 
39
  percent = prob * 100
40
  top += percent
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")
48
  model = init_model()