from core.model_loader import load_model
from core.tokenizer_loader import load_tokenizer
from core.dataset_loader import load_dataset_single
dataset = load_dataset_single()
from core.trainer import build_trainer

MODEL_NAME = "tiiuae/falcon-7b-instruct"
HUB_ID = "Mauricio-100/agent-ai"

print("🚀 Lancement de l’entraînement de Gopu Agent...")

model = load_model(MODEL_NAME)
tokenizer = load_tokenizer(MODEL_NAME)
dataset = load_multiple_datasets()

trainer = build_trainer(model, dataset, tokenizer, HUB_ID)
trainer.train()
trainer.push_to_hub()

print("✅ Modèle entraîné et publié sur Hugging Face !")