Update app.py

app.py

@@ -107,8 +107,7 @@ def inf(inpt):
     embedding = sentencemodel.encode(TEXT, convert_to_tensor=True)
     INPUT = torch.cat((probs, embedding))
     output = F.softmax(model(INPUT.view(1, -1)), dim=1)
-
-    if output[0][0] > output[0][1]:
+    if not output[0][1] > 0.62:
         return "Not toxic " + str(output[0][0])
     else:
         return "Toxic! " + str(output[0][1])