Spaces:
Runtime error
Runtime error
Kirill commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ 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):
|
|
@@ -70,4 +70,4 @@ iface = gr.Interface(
|
|
| 70 |
title="Определение спама в русскоязычных текстах"
|
| 71 |
)
|
| 72 |
|
| 73 |
-
iface.launch()
|
|
|
|
| 50 |
outputs = model(input_ids, attention_mask=attention_mask).logits
|
| 51 |
pred = torch.sigmoid(outputs).cpu().numpy()[0][0]
|
| 52 |
|
| 53 |
+
result = "Спам" если pred >= 0.5 еще "Не спам"
|
| 54 |
return result
|
| 55 |
|
| 56 |
def predict_spam(text, model_choice):
|
|
|
|
| 70 |
title="Определение спама в русскоязычных текстах"
|
| 71 |
)
|
| 72 |
|
| 73 |
+
iface.launch()
|