babi

src/hf.py

@@ -2,7 +2,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)
+tokenizer = BertTokenizer.from_pretrained(model_path, local_files_only=True)
 # model = BertForSequenceClassification.from_pretrained(model_path)
 model = AutoModelForSequenceClassification.from_pretrained(model_path)