Update app.py

app.py

@@ -61,10 +61,13 @@ transformers.utils.logging.set_verbosity_error()
 task_name = 'sst2'
 model_type = 'transkimer'
 
+label_list = processor.get_labels()
+label_to_id = {v: i for i, v in enumerate(label_list)}
+
 # Load pretrained model and tokenizer
 model_path_key =  f'{model_type}_{task_name}_not_pad'
 model_path = model_path_dict[model_path_key]
-config = AutoConfig.from_pretrained(model_path, num_labels=num_labels, finetuning_task=task_name)
+config = AutoConfig.from_pretrained(model_path, num_labels=len(label_list), finetuning_task=task_name)
 tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', use_fast=True)
 model = TranskimerForSequenceClassification.from_pretrained(model_path,from_tf=bool(".ckpt" in model_path),config=config,)
 
@@ -72,9 +75,6 @@ model = TranskimerForSequenceClassification.from_pretrained(model_path,from_tf=b
 sentence1_key, sentence2_key = task_to_keys[task_name]
     
 processor = processors[task_name]()
-label_list = processor.get_labels()
-    
-label_to_id = {v: i for i, v in enumerate(label_list)}
 
 padding = False