File size: 1,366 Bytes
a42610b 7005f78 a42610b 7005f78 a42610b 7005f78 a42610b 7005f78 |
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 |
from DistilBERT import model_DB
import streamlit as st
from transformers import DistilBertTokenizer, DistilBertModel
import logging
logging.basicConfig(level=logging.ERROR)
import torch
MAX_LEN = 100
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased', truncation=True, do_lower_case=True)
def sentiment_analysis_DB(input):
inputs = tokenizer.encode_plus(
input,
None,
add_special_tokens=True,
max_length=MAX_LEN,
pad_to_max_length=True,
return_token_type_ids=True
)
ids = inputs['input_ids']
mask = inputs['attention_mask']
token_type_ids = inputs["token_type_ids"]
output = model_DB(ids, mask, token_type_ids)
final_outputs = np.array(output)
final_outputs = final_outputs[0]
if final_outputs == True:
result = 1
else:
result = 0
return result
# Streamlit app
st.title("Sentiment Analysis App")
# User input
user_input = st.text_area("Enter some text:")
# Button to trigger sentiment analysis
if st.button("Analyze Sentiment"):
# Perform sentiment analysis
result = sentiment_analysis_DB(user_input)
# Display result
if result == 1:
st.success("Positive sentiment detected!")
else:
st.error("Negative sentiment detected.") |