from transformers import GPT2Tokenizer, GPT2LMHeadModel

tokenizer = GPT2Tokenizer.from_pretrained("Muyumba/oprimus_ai")
model = GPT2LMHeadModel.from_pretrained("Muyumba/oprimus_ai")

# Génération de texte
input_text = "Bonjour, comment allez-vous?"
input_ids = tokenizer.encode(input_text, return_tensors="pt")
output = model.generate(input_ids, max_length=100, do_sample=True)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(generated_text)

Détails du modèle

Modèle de base: GPT-2
Langue: Français
Type: Génération de texte