final_1

app.py

@@ -110,7 +110,7 @@ def predict_label(sentence, tokenizer, model1, model2, device):
         "score": probabilities[common_label_map[predicted_label]]
     }
 
-    return json.dumps(response)
+    return response
 
 
 # Streamlit app
@@ -123,4 +123,4 @@ sentence = st.text_area("Enter the sentence to predict your mental illness state
 if st.button('Predict'):
     # Predict label
     predicted_response = predict_label(sentence, tokenizer, mi_model, roberta_model, device)
-    st.write("results:", predicted_response)
+    st.json(predicted_response)