Update app.py

app.py

@@ -49,4 +49,5 @@ if user_input and button :
                                  return_tensors="tf")
     output = model(predict_input)[0]
     prediction_value = tf.argmax(output, axis=1).numpy()[0]
+    st.write("Logits: ",prediction_value)
     st.write("Prediction: ",d[prediction_value])