ashrestha commited on
Commit
db840ed
1 Parent(s): adf8ee2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,7 +15,9 @@ if submit_button:
15
 
16
  labels = list(l.strip() for l in input_label.split(','))
17
  pred = classifier(input_text, labels, multi_class=True)
 
 
18
 
19
- st.success(pred['labels'][0:1])
20
 
21
  # st.markdown(pred)
 
15
 
16
  labels = list(l.strip() for l in input_label.split(','))
17
  pred = classifier(input_text, labels, multi_class=True)
18
+
19
+ out = f"Top predicted labels are {', '.join(p for p in pred['labels'][0:2])}"
20
 
21
+ st.success(out)
22
 
23
  # st.markdown(pred)