File size: 918 Bytes
a4c6c31 9d5a490 47ea938 9d5a490 a4c6c31 9d5a490 86bc0ed 9d5a490 a4c6c31 4637a97 a4c6c31 |
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 |
import streamlit as st
from transformers import pipeline
def sentiment_funtion(text):
model = pipeline("sentiment-analysis", model="stevhliu/my_awesome_model")
output = model(text)[0]
label = output["label"]
score = output["score"]
return label , score
def main():
user_input = st.text_area("Input Text Here")
if st.button("Sentiment"):
label , score = sentiment_funtion(user_input)
if label == "LABEL_1":
label = "Positive"
postive_score = score
negative_score = 1 - score
else:
label = "Negative"
postive_score = 1 - score
negative_score = score
st.write("Sentiment:")
st.write(f"Sentiment Label: {label}")
st.write(f"Positive Score: {postive_score:.2f}")
st.write(f"Negative Score: {negative_score:.2f}")
if __name__ == "__main__":
main() |