Update README.md

README.md

@@ -20,6 +20,7 @@ model-index:
 Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
 
 >>> print(tokenizer.decode(tokens[0]))
+
 <s><|im_start|> system
 You are a helpful assistant.<|im_end|> 
 <|im_start|> user
@@ -28,6 +29,25 @@ En 1975, la localité comptait 90 habitants, des Guiziga et lors du recensement
 <|im_start|> assistant
 When the town had 90 inhabitants in 1975, it was called Guizaga and during the census of 2005, there were x inhabitants.<|im_end|> 
 </s>
+
+
+>>> chat_input = "<|im_start|> system\nYou are a helpful assistant.<|im_end|> \n<|im_start|> user\nCorrige les fautes dans ce texte : \nEn 1975, la localité comptait 90 habitant, des Guiziga et lors du recensement de 2005, on y a dénombrer 56 habitants.<|im_end|> \n<|im_start|> assistant\n"
+
+>>> inputs = tokenizer(chat_input, return_tensors="pt").to(model.device)
+
+>>> tokens = model.generate(**inputs, **generation_args)
+Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
+
+>>> print(tokenizer.decode(tokens[0]))
+<s><|im_start|> system
+You are a helpful assistant.<|im_end|> 
+<|im_start|> user
+Corrige les fautes dans ce texte : 
+En 1975, la localité comptait 90 habitant, des Guiziga et lors du recensement de 2005, on y a dénombrer 56 habitants.<|im_end|> 
+
+<|im_start|> assistant
+ En 1975, la commune comptait 90 habitants dont des Guizigas et au recensement de 2005, elle en compte 56.<|im_end|> 
+</s>
 >>> 
 ```