Fix a bug

eval_utils.py

@@ -25,8 +25,8 @@ def predict_top_p(model, tokenizer, text, P=0.9):
             break
 
     if (
-        str(model.config.to_dict()["id2label"]["0"]) == "LABEL_0"
-        or str(model.config.to_dict()["id2label"]["0"]) == "Algeria"
+        str(model.config.to_dict()["id2label"][0]) == "LABEL_0"
+        or str(model.config.to_dict()["id2label"][0]) == "Algeria"
     ):
         return [DIALECTS[i] for i, p in enumerate(predictions) if p == 1]
     else: