Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,8 @@ def predict_spam_spamns(text):
|
|
| 50 |
outputs = model(input_ids, attention_mask=attention_mask).logits
|
| 51 |
pred = torch.sigmoid(outputs).cpu().numpy()[0][0]
|
| 52 |
|
| 53 |
-
result = "Спам"
|
|
|
|
| 54 |
return result
|
| 55 |
|
| 56 |
def predict_spam(text, model_choice):
|
|
|
|
| 50 |
outputs = model(input_ids, attention_mask=attention_mask).logits
|
| 51 |
pred = torch.sigmoid(outputs).cpu().numpy()[0][0]
|
| 52 |
|
| 53 |
+
result = "Спам" if pred >= 0.5 else "Не спам"
|
| 54 |
+
|
| 55 |
return result
|
| 56 |
|
| 57 |
def predict_spam(text, model_choice):
|