Update app.py

app.py

@@ -180,7 +180,7 @@ if st.button("Process"):
                     classes = []
                     with st.spinner(text="creating output file"):
                         for x in stqdm(range(len(text))):
-                            tf_o = tf.nn.softmax(tf_outputs["logits"][x], axis=-1)
+                            tf_o = softmax(tf_outputs["logits"][x], axis=-1)
                             label = np.argmax(tf_o, axis=0)
                             keys = model_classes
                             classes.append(keys.get(label))
@@ -221,7 +221,7 @@ if st.button("Process"):
                     classes = []
                     with st.spinner(text="creating output file"):
                         for x in stqdm(range(len(text))):
-                            tf_o = tf.nn.softmax(tf_outputs["logits"][x], axis=-1)
+                            tf_o = softmax(tf_outputs["logits"][x], axis=-1)
                             label = np.argmax(tf_o, axis=0)
                             keys = model_classes
                             classes.append(keys.get(label))
@@ -242,4 +242,4 @@ if st.button("Process"):
             )
             st.info("Text column must have amazon reviews", icon="ℹ️")
         except BaseException as e:
-            logging.exception("An exception was occured")
+            logging.exception("An exception was occurred")