Update app.py

app.py

@@ -1,9 +1,22 @@
 import streamlit as st
+import numpy as np
+import tensorflow as tf
 
 x = st.header("Welcome to the STEM NLP application!")
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+model = TFDistilBertForSequenceClassification.from_pretrained("kaixinwang/NLP")
+
+MODEL_NAME_1 = 'distilbert-base-uncased'
+tokenizer = DistilBertTokenizer.from_pretrained(MODEL_NAME_1)
 
 x = st.text_input("Type in your review here:")
-st.write("Your review is:", x)
+st.write("Your review is:", x)
+encoding = tokenizer(x, truncation=True, padding=True)
+encoded = tf.data.Dataset.from_tensor_slices((dict(encoding), 1)
+preds = model.predict(encoded.batch(1)).logits  
+prob = tf.nn.softmax(preds, axis=1).numpy()  
+prob_max = np.argmax(prob, axis=1)
+st.write("Sentiment:", prob_max, "Score:", prob[prob_max])
+
+# x = st.slider('Select a value')
+# st.write(x, 'squared is', x * x)