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()