Update app.py

app.py

@@ -6,6 +6,11 @@ def install(package):
 
 install("tensorflow")
 install("numpy")
+install(transformers)
+
+import transformers
+from transformers import DistilBertTokenizer
+from transformers import TFDistilBertForSequenceClassification
 
 import streamlit as st
 import numpy as np
@@ -21,7 +26,7 @@ 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))
+encoded = tf.data.Dataset.from_tensor_slices((dict(encoding), np.ones(1)))
 preds = model.predict(encoded.batch(1)).logits  
 prob = tf.nn.softmax(preds, axis=1).numpy()  
 prob_max = np.argmax(prob, axis=1)