beanbox-apis

Running

johnpaulbin commited on Dec 17, 2023

Commit

fcf5457

1 Parent(s): 402700e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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