kandy/Sentiment

1import joblib2import pandas as pd3import streamlit as st  #ha4import numpy as np5import sklearn6from sklearn.feature_extraction.text import TfidfVectorizer7 8from sklearn.naive_bayes import MultinomialNB9 10model = joblib.load('model_nb.joblib')11tfid = joblib.load('tfidf_nb.joblib')12 13def main():14    st.title('Input your text 4-5 words it will tell your mood!!!  example "i fell romantic", "i love you"')15    with st.form('Hello World'):16        text = st.text_input('Input word here.')17        clicked = st.form_submit_button('START')18        if clicked:19            mood=[]20            textz = [text]21            X = tfid.transform(textz)22            result = model.predict(X)23            if result[0] == 0 :24                mood = 'SAD'25            elif result[0] == 1:26                mood = 'JOY'27            elif result[0] == 2:28                mood = 'LOVE'  29            elif result[0] == 3:30                mood = 'anger'31            elif result[0] == 4:32                mood = 'fear'33            else:34                mood = 'surprise'                         35            36            st.success('Your mood is '+ mood)37        38       39if __name__ == '__main__':40    main()