Spaces:

non2013
/

SincereQuestions

Sleeping

non2013 commited on Oct 22, 2024

Commit

797ba3e

1 Parent(s): c617a29

edit padding to not show

Files changed (1) hide show

app.py CHANGED Viewed

@@ -103,7 +103,7 @@ class DeBERTaV3Model:
         label = "Insincere" if (F.softmax(preds[0], dim=1)[:, 1]>0.4878) else "Sincere"
         probs = {
             "Probability": float(F.softmax(preds[0], dim=1)[:, 1]),
-            "Sequence": test_tensor
         }
         return label, probs

         label = "Insincere" if (F.softmax(preds[0], dim=1)[:, 1]>0.4878) else "Sincere"
         probs = {
             "Probability": float(F.softmax(preds[0], dim=1)[:, 1]),
+            "Sequence": test_tensor[test_tensor != 0]
         }
         return label, probs