| # update_model_config.py | |
| from transformers import AutoConfig, AutoModelForSequenceClassification | |
| model_id = "KenLumod/bert-fake-news-detector" | |
| def update_label_mapping(): | |
| # Load existing config and model | |
| config = AutoConfig.from_pretrained(model_id) | |
| model = AutoModelForSequenceClassification.from_pretrained(model_id) | |
| # Update label mapping | |
| config.id2label = {0: "Fake News", 1: "Real News"} | |
| config.label2id = {v: k for k, v in config.id2label.items()} | |
| # Save updated files | |
| model.save_pretrained(model_id) | |
| config.save_pretrained(model |