Update app.py

app.py

@@ -40,7 +40,7 @@ MODEL_CLASSES = {
     "albert": (AlbertConfig, AlbertForSequenceClassification, AlbertTokenizer),
 }
 
-model = 'albert'
+model_name = 'albert'
 dataset = 'STS-B'
 task_name = f'{dataset}'.lower()
 if task_name not in processors:
@@ -53,14 +53,14 @@ num_labels = len(label_list)
 output_dir = f'./outputs/train/{model}/{dataset}'
 data_dir = f'./glue_data/{dataset}'
 
-config_class, model_class, tokenizer_class = MODEL_CLASSES[model]
+config_class, model_class, tokenizer_class = MODEL_CLASSES[model_name]
 tokenizer = tokenizer_class.from_pretrained(output_dir, do_lower_case=True)
-model = model_class.from_pretrained(best_model_path[f'{model}_{dataset}'])
+model = model_class.from_pretrained(best_model_path[f'{model_name}_{dataset}'])
 
 exit_type='patience'
 exit_value=3
 
-classifier = MyClassifier(model,tokenizer,label_list,output_mode,exit_type,exit_value,model)
+classifier = MyClassifier(model,tokenizer,label_list,output_mode,exit_type,exit_value,model_name)
 
 def greet(text,text2,exit_pos):
     text_input = [(text,text2)]