NeuroSpaceX commited on
Commit
46f08aa
·
verified ·
1 Parent(s): 5f5dede

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "Спам" if pred >= 0.5 else "Не спам"
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()