Update app.py

app.py

@@ -37,7 +37,7 @@ def predict_ner(model, tokenizer, text, label_map, device):
 
     # Predict NER labels
     with torch.no_grad():
-        logits = model(input_ids=input_ids, attention_mask=attention_mask, labels=None, labels_mask=None, valid_mask=None)
+        logits = model(inputs_ids=input_ids, labels=None, labels_mask=None, valid_mask=None)
 
     # Convert logits to label IDs
     label_ids = torch.argmax(logits, dim=2).detach().cpu().numpy()