File size: 475 Bytes
291793f c12c8cd a681e0a c4020ea a681e0a ff694a9 a681e0a 291793f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# Use a pipeline as a high-level helper
from transformers import pipeline
import streamlit as st
pipe = pipeline("text-classification", model="mavinsao/mi-roberta-base-finetuned-mental-illness")
# Streamlit app
st.title('Mental Illness Prediction')
# Input text area for user input
sentence = st.text_area("Enter the long sentence to predict your mental illness state:")
# Prediction button
if st.button('Predict'):
st.write("Predicted labels:", pipe(sentence))
|