Update app.py

app.py

@@ -29,14 +29,18 @@ def predict_spam_deberta(text):
     model = model_instances["RUSpam/spam_deberta_v4"]
 
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
+    input_ids = inputs['input_ids'].to(device)
+    attention_mask = inputs['attention_mask'].to(device)
+
     with torch.no_grad():
-        outputs = model(**inputs)
-        logits = outputs.logits
-        predicted_class = torch.argmax(logits, dim=1).item()
-    
-    result = "Спам" if predicted_class == 1 else "Не спам"
+        outputs = model(input_ids, attention_mask=attention_mask).logits
+        pred = torch.sigmoid(outputs).cpu().numpy()[0][0]
+
+    is_spam = int(pred >= 0.5)
+    result = "Спам" if is_spam == 1 else "Не спам"
     return result
 
+
 def predict_spam_spamns(text):
     tokenizer = tokenizers["RUSpam/spamNS_v1"]
     model = model_instances["RUSpam/spamNS_v1"]