|
|
import streamlit as st |
|
|
|
|
|
import pandas as pd |
|
|
import joblib |
|
|
|
|
|
st.set_page_config(page_title="Stress Classification",layout="wide") |
|
|
|
|
|
st.markdown("<h1 style='font-size:40px;color:darkred;text-align:center'>Stress Categorization Using BERT Algorithm</h1>",unsafe_allow_html=True) |
|
|
|
|
|
|
|
|
st.markdown("<h2 style='font-size:30px;color:black;'>Enter your concern</h2>",unsafe_allow_html=True) |
|
|
posts = st.text_area("") |
|
|
|
|
|
if st.button("Predict"): |
|
|
vector = joblib.load("stress.h5") |
|
|
|
|
|
predictions = vector.transform([posts]) |
|
|
|
|
|
model = joblib.load("stress_model.h5") |
|
|
|
|
|
pred = model.predict(predictions) |
|
|
|
|
|
if pred[0]=="Financial Stress": |
|
|
st.success("This Posts suggests that person has Financial Stress.") |
|
|
elif pred[0]=="Health Stress": |
|
|
st.success("This Posts suggests that person has Health Stress.") |
|
|
elif pred[0] == "Other Stress": |
|
|
st.success("This Posts suggests that person has Other Type Of Stress.") |
|
|
elif pred[0] =="Work Stress": |
|
|
st.success("This Posts suggests that person has Work Stress") |
|
|
|
|
|
|
|
|
|
