fixed id2label issue

app.py

@@ -17,6 +17,14 @@ HF_TOKEN = os.environ["hf_read"]
 SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment", 2: "Positive"}
 LANGUAGES = ["Czech", "English", "French", "German", "Hungarian", "Polish", "Slovakian"]
 
+id2label = {
+    0: "Anger",
+    1: "Fear",
+    2: "Disgust",
+    3: "Sadness",
+    4: "Joy",
+    5: "None of Them"
+}
 def load_spacy_model(model_name="xx_sent_ud_sm"):
     try:
         model = spacy.load(model_name)
@@ -58,7 +66,7 @@ def predict(text, model_id, tokenizer_id):
     return probs
 
 def get_most_probable_label(probs):
-    label = model.config.id2label[probs.argmax()]
+    label = id2label[probs.argmax()]
     probability = f"{round(100 * probs.max(), 2)}%"
     return [sentence, label, probability]