babi

src/hf.py

@@ -4,7 +4,7 @@ from transformers import BertTokenizer, BertForSequenceClassification,TextClassi
 # Load tokenizer and model from the fine-tuned directory
 model_path = './intent_classification/TinyBERT_106_V2'  # can try other checkpoints
 
-tokenizer = BertTokenizer.from_pretrained(model_path, local_files_only=True)
+tokenizer = BertTokenizer.from_pretrained('KhairulAmirinUM/Advisor_AI_BERT')
 # model = BertForSequenceClassification.from_pretrained(model_path)
 model = AutoModelForSequenceClassification.from_pretrained(model_path, local_files_only=True)
 print(os.path.exists(model_path))