Update tasks/text.py

tasks/text.py

@@ -24,6 +24,15 @@ BINARY_MODEL = "Medissa/Roberta_Binary"
 TOKENIZER_MULTI_CLASS = "crarojasca/TaxonomyAugmentedCARDS"
 MULTI_CLASS_MODEL = "Medissa/Deberta_Taxonomy"
 
+ID2LABEL = {
+ 0: '1_not_happening',
+ 1: '2_not_human',
+ 2: '3_not_bad',
+ 3: '4_solutions_harmful_unnecessary',
+ 4: '7_fossil_fuels_needed',
+ 5: '5_science_unreliable',
+ 6: '6_proponents_biased'}
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 from fastapi import APIRouter