Remove itos

app.py

@@ -123,6 +123,6 @@ def predict_ner(sentence: str, model, id2label: dict, device="cpu"):
 
 
 sentence = "ذهب احمد الى السوق"
-id2label = {i: s for i, s in enumerate(label_vocab.itos)}
-pairs = predict_ner(sentence, model, id2label, device="cpu")
+# id2label = {i: s for i, s in enumerate(label_vocab.itos)}
+pairs = predict_ner(sentence, model, label_vocab, device="cpu")
 print(pairs)