pipelinetest

app.py

@@ -1,8 +1,16 @@
 import streamlit as st
-import numpy as np
+from transformers import pipeline
+# import numpy as np
 
-st.text_input("Input Text Here:", "I really like the color of your car!")
+classifier = pipeline(task="sentiment-analysis")
 
-option = st.selectbox('What pre-trained model would you like for your sentiment analysis?',('TensorFlow', 'PyTorch', 'JAX'))
+textIn = st.text_input("Input Text Here:", "I really like the color of your car!")
+
+option = st.selectbox('Which pre-trained model would you like for your sentiment analysis?',('TensorFlow', 'PyTorch', 'JAX'))
 
 st.write('You selected:', option)
+
+# pipeline
+preds = classifier(textIn)
+preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
+st.write('Input text is ', preds[0]['label'], ' with a confidence of ', preds[0]['score'])