Update app.py

app.py

@@ -12,7 +12,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), 1))
 preds = model.predict(encoded.batch(1)).logits  
 prob = tf.nn.softmax(preds, axis=1).numpy()  
 prob_max = np.argmax(prob, axis=1)