LikoKIko commited on
Commit
7fe4035
·
verified ·
1 Parent(s): fe5233f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def check(txt: str) -> str:
40
  ).to(KDevice)
41
 
42
  prob = torch.sigmoid(model(**batch).logits).item()
43
- label = 1 if prob >= KThreshold else 0
44
  return f"Prob: {prob:.4f} | Label: {label} (cutoff={KThreshold})"
45
 
46
  # ui
 
40
  ).to(KDevice)
41
 
42
  prob = torch.sigmoid(model(**batch).logits).item()
43
+ label = "לא פוגעני" if prob >= KThreshold else "פוגעני"
44
  return f"Prob: {prob:.4f} | Label: {label} (cutoff={KThreshold})"
45
 
46
  # ui