boltuix commited on
Commit
af54376
·
verified ·
1 Parent(s): ac6bcc0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -322,7 +322,7 @@ pip install transformers torch pandas scikit-learn tqdm
322
  from transformers import AutoModelForSequenceClassification
323
 
324
  # 📥 Load the fine-tuned intent classification model
325
- model = AutoModelForSequenceClassification.from_pretrained("bert-local")
326
 
327
  # 🏷️ Extract the ID-to-label mapping dictionary
328
  label_mapping = model.config.id2label
 
322
  from transformers import AutoModelForSequenceClassification
323
 
324
  # 📥 Load the fine-tuned intent classification model
325
+ model = AutoModelForSequenceClassification.from_pretrained("boltuix/bert-local")
326
 
327
  # 🏷️ Extract the ID-to-label mapping dictionary
328
  label_mapping = model.config.id2label