Spaces:
Runtime error
Runtime error
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()
|