more_stuff

app.py

@@ -59,7 +59,7 @@ if not st.session_state.filled:
                 logits = predictions.logits
                 sigmoid = torch.nn.Sigmoid()
                 probs = sigmoid(logits.squeeze().cpu())
-                predictions = np.zeros(probs.shape)
+                predicts = np.zeros(probs.shape)
                 predictions[np.where(probs >= 0.5)] = 1
                 predicted_labels = [st.session_state.id2label[idx] for idx, label in enumerate(predictions) if label == 1.0]
             log = []
@@ -106,7 +106,7 @@ if not st.session_state.filled:
             else:
                 log = [0] * 6
                 log[1] = text
-                if max(predictions) == 0:
+                if max(predicts) == 0:
                     log[0] = 0
                     log[2] = ("NO TOXICITY")
                     log[3] = (f"{100 - round(probs[0].item() * 100, 1)}%")
@@ -116,7 +116,7 @@ if not st.session_state.filled:
                     log[0] = 1
                     _max = 0
                     _max2 = 2
-                    for i in range(1, len(predictions)):
+                    for i in range(1, len(predicts)):
                         if probs[i].item() > probs[_max].item():
                             _max = i
                         if i > 2 and probs[i].item() > probs[_max2].item():