Update app.py

app.py

@@ -1,17 +1,30 @@
 import streamlit as st
 from transformers import pipeline
 
-def summarize_text(text):
+def sentiment_funtion(text):
     model = pipeline("text-classification", model = "cross-encoder/qnli-electra-base")
-    text_input = model(text)
-    return text_input
+    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"):
-        summary = summarize_text(user_input)
+        label , score = sentiment_funtion(user_input)
+        if label == "LABEL_0":
+            label = "Positive"
+            postive_score = score
+            negative_score = 1 - score
+        else:
+            label = "Negative"
+            postive_score = 1 - score
+            negative_score = score
+            
         st.write("Sentiment:")
-        st.write(summary)
+        st.write(label)
+        st.write(postive_score)
+        st.write(negative_score)
 
 if __name__ == "__main__":
     main()