push_model.py

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model_path = "models/fine_tuned_model"  # Chemin vers ton modèle fine-tuné
repo_name = "agnews-finetuned-bert"     # Nom public sur Hugging Face

# Charger le modèle et le tokenizer
model = AutoModelForSequenceClassification.from_pretrained(model_path)
tokenizer = AutoTokenizer.from_pretrained(model_path)

# Uploader
model.push_to_hub(repo_name)
tokenizer.push_to_hub(repo_name)