Spaces:

johnpaulbin
/

beanbox-toxicity

Runtime error

johnpaulbin commited on Dec 18, 2023

Commit

24a888d

1 Parent(s): eabbd6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,7 +131,7 @@ def translateverbose():
     INPUT = torch.cat((probs, embedding))
     output = F.softmax(model(INPUT.view(1, -1)), dim=1)
-    if output[0][1] > 0.68:
         output = "true" + str(output[0][1])
     else:
         output = "false" + str(output[0][0])

     INPUT = torch.cat((probs, embedding))
     output = F.softmax(model(INPUT.view(1, -1)), dim=1)
+    if output[0][1] > 0.4:
         output = "true" + str(output[0][1])
     else:
         output = "false" + str(output[0][0])