Update README.md

README.md

@@ -52,10 +52,18 @@ This is a small multilingual language model based on a Transformer architecture
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Duino/Darija-GPT") # Replace with your actual HF Hub Repo ID
-model = AutoModelForCausalLM.from_pretrained("Duino/Darija-GPT") # Replace with your actual HF Hub Repo ID
+tokenizer = AutoTokenizer.from_pretrained("Duino/Darija-GPT") 
+model = AutoModelForCausalLM.from_pretrained("Duino/Darija-GPT") 
 
-# Example usage (text generation or language modeling code would go here)
+prompt_text = "هذا نموذج لغوي صغير" # Example prompt in Arabic/Darija
+input_ids = tokenizer.encode(prompt_text, return_tensors="pt").to(model.device)
+
+# Generate text (adjust max_length, temperature, top_p as needed)
+output = model.generate(input_ids, max_new_tokens=50, temperature=0.9, top_p=0.9)
+
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+print("Prompt:", prompt_text)
+print("Generated text:", generated_text)
 ```
 
 ## Training Plot