Update app.py

app.py

@@ -33,11 +33,11 @@ def predict_spam_deberta(text):
     attention_mask = inputs['attention_mask'].to(device)
 
     with torch.no_grad():
-        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 "Не спам"
+        outputs = model(**inputs)
+        logits = outputs.logits
+        predicted_class = torch.argmax(logits, dim=1).item()
+    
+    result = "Спам" if predicted_class == 1 else "Не спам"
     return result