Update app.py

app.py

@@ -56,6 +56,7 @@ def predict_text(params):
         pred = bert_model.predict([input_ids, token_type_ids, attention_mask])
         logits = pred.logits
         pred_label = tf.argmax(logits, axis=1).numpy()[0]
+        print(f"{logits}\t{pred_label}")
         if not pred_label:
             predict_label = 7