config jit issue

interfaces/emotion.py

@@ -7,7 +7,7 @@ from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer
 from huggingface_hub import HfApi
 
-from label_dicts import MANIFESTO_LABEL_NAMES
+from label_dicts import EMOTION6_LABEL_NAMES
 
 from .utils import is_disk_full, release_model
 
@@ -54,7 +54,7 @@ def predict(text, model_id, tokenizer_id):
     release_model(model, model_id)
 
     probs = torch.nn.functional.softmax(logits, dim=1).cpu().numpy().flatten()
-    output_pred = {model.config.id2label[i]: probs[i] for i in np.argsort(probs)[::-1]}
+    output_pred = {EMOTION6_LABEL_NAMES[i]: probs[i] for i in np.argsort(probs)[::-1]}
     output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
     return output_pred, output_info
 

interfaces/manifesto.py

@@ -7,7 +7,7 @@ from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer
 from huggingface_hub import HfApi
 
-from label_dicts import MANIFESTO_LABEL_NAMES
+from label_dicts import MANIFESTO_LABEL_NAMES, MANIFESTO_NUM_DICT
 
 from .utils import is_disk_full, release_model
 
@@ -53,7 +53,7 @@ def predict(text, model_id, tokenizer_id):
     release_model(model, model_id)
 
     probs = torch.nn.functional.softmax(logits, dim=1).cpu().numpy().flatten()
-    output_pred = {f"[{model.config.id2label[i]}] {MANIFESTO_LABEL_NAMES[int(model.config.id2label[i])]}": probs[i] for i in np.argsort(probs)[::-1]}
+    output_pred = {f"[{model.config.id2label[i]}] {MANIFESTO_LABEL_NAMES[int(MANIFESTO_NUM_DICT[i])]}": probs[i] for i in np.argsort(probs)[::-1]}
     output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
     return output_pred, output_info
 

label_dicts.py

@@ -549,6 +549,66 @@ CAP_MIN_MEDIA_NUM_DICT = {0: 100,
      217: 30,
      218: 31,
      219: 99}
+     
+MANIFESTO_NUM_DICT = {
+    0: 0,
+    1: 101,
+    2: 102,
+    3: 103,
+    4: 104,
+    5: 105,
+    6: 106,
+    7: 107,
+    8: 108,
+    9: 109,
+    10: 110,
+    11: 201,
+    12: 202,
+    13: 203,
+    14: 204,
+    15: 301,
+    16: 302,
+    17: 303,
+    18: 304,
+    19: 305,
+    20: 401,
+    21: 402,
+    22: 403,
+    23: 404,
+    24: 405,
+    25: 406,
+    26: 407,
+    27: 408,
+    28: 409,
+    29: 410,
+    30: 411,
+    31: 412,
+    32: 413,
+    33: 414,
+    34: 415,
+    35: 416,
+    36: 501,
+    37: 502,
+    38: 503,
+    39: 504,
+    40: 505,
+    41: 506,
+    42: 507,
+    43: 601,
+    44: 602,
+    45: 603,
+    46: 604,
+    47: 605,
+    48: 606,
+    49: 607,
+    50: 608,
+    51: 701,
+    52: 702,
+    53: 703,
+    54: 704,
+    55: 705,
+    56: 706
+}
 
 CAP_MIN_LABEL_NAMES = {
     # 1. Macroeconomics
@@ -905,6 +965,16 @@ ONTOLISST_LABEL_NAMES = {
     14: 'Administration',
     15: 'COVID19'
 }
+
+EMOTION6_LABEL_NAMES = {
+    0: "Anger",
+    1: "Fear",
+    2: "Disgust",
+    3: "Sadness",
+    4: "Joy",
+    5: "None of Them"
+}
+
 EMOTION9_LABEL_NAMES = {
     0: "Anger",
     1: "Fear",