beanbox-apis

Running

johnpaulbin commited on Dec 16, 2023

Commit

402700e

1 Parent(s): aea79b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     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"
     return output