Spaces:
Paused
Paused
File size: 2,254 Bytes
1b34781 f0acac6 93eddbd 71cc21f 2d06877 71cc21f abbd961 93eddbd db7041a 93eddbd 4da66d5 db7041a f0acac6 db7041a 93eddbd db7041a 93eddbd db7041a 93eddbd db7041a 93eddbd db7041a 93eddbd db7041a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import streamlit as st
from functions_preprocess import LinguisticPreprocessor, download_if_non_existent, CNN
import pickle
import nltk
nltk.download('stopwords')
nltk.download('punkt')
download_if_non_existent('corpora/stopwords', 'stopwords')
download_if_non_existent('taggers/averaged_perceptron_tagger', 'averaged_perceptron_tagger')
download_if_non_existent('corpora/wordnet', 'wordnet')
#################################################################### Streamlit interface
st.title("Movie Reviews: An NLP Sentiment analysis")
#################################################################### Cache the model loading
@st.cache_data()
def load_model():
model_pkl_file = "sentiment_model.pkl"
with open(model_pkl_file, 'rb') as file:
model = pickle.load(file)
return model
def load_cnn():
model = CNN(len(vocab), 300, 128, [3, 8], 0.5, 2)
model.load_state_dict(torch.load('model_cnn.pkl'))
model.eval()
return model
def predict_sentiment(text, model):
processor.transform(text)
prediction = model.predict([text])
return prediction
model_1 = load_model()
model_2 = load_cnn()
processor = LinguisticPreprocessor()
############################################################# Text input
with st.expander("Model 1: SGD Classifier"):
st.markdown("Give it a go by writing a positive or negative text, and analyze it!")
# Text input inside the expander
user_input = st.text_area("Enter text here...")
if st.button('Analyze'):
# Displaying output
result = predict_sentiment(user_input, model_1)
if result >= 0.5:
st.write('The sentiment is: Positive 😀')
else:
st.write('The sentiment is: Negative 😞')
with st.expander("Model 2: CNN Sentiment analysis"):
st.markdown("Give it a go by writing a positive or negative text, and analyze it!")
# Text input inside the expander
user_input = st.text_area("Enter text here...")
if st.button('Analyze'):
# Displaying output
result = predict_sentiment(user_input, model_2)
if result >= 0.5:
st.write('The sentiment is: Positive 😀')
else:
st.write('The sentiment is: Negative 😞')
st.caption("Por @efeperro.") |