updated 5k model version

app.py

@@ -5,7 +5,7 @@ from transformers import BertTokenizer, BertForSequenceClassification
 
 label_dict = {"Urgency": 0, "Not Dark Pattern": 1, "Scarcity": 2, "Misdirection": 3, "Social Proof": 4, "Obstruction": 5, "Sneaking": 6, "Forced Action": 7}
 model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=len(label_dict))
-fine_tuned_model_path = "models/finetuned_BERT_epoch_5.model"
+fine_tuned_model_path = "models/finetuned_BERT_5k_epoch_5.model"
 model.load_state_dict(torch.load(fine_tuned_model_path, map_location=torch.device('cpu')))
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased', do_lower_case=True)