Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
43 |
|
44 |
chart_data = pd.DataFrame([top_probs], columns=top_cats)
|
45 |
-
st.
|
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()
|