mavinsao commited on
Commit
0b2c300
·
verified ·
1 Parent(s): dd1c6f2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -122,4 +122,4 @@ sentence = st.text_area("Enter the sentence to predict your mental illness state
122
  if st.button('Predict'):
123
  # Predict label
124
  predicted_response = predict_label(sentence, tokenizer, mi_model, roberta_model, device)
125
- st.json(predicted_response)
 
122
  if st.button('Predict'):
123
  # Predict label
124
  predicted_response = predict_label(sentence, tokenizer, mi_model, roberta_model, device)
125
+ st.write("results:", predicted_response)