app.py

!pip install tensorflow
!pip install numpy

import streamlit as st
import numpy as np
import tensorflow as tf

x = st.header("Welcome to the STEM NLP application!")

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