Update README.md

README.md

@@ -55,16 +55,16 @@ To get started with the model, follow the steps below:
 
 ```python
 # Example code
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+>>> from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-model_name = "Saaidtaoussi/AraT5_Darija_to_MSA"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+>>> model_name = "Saaidtaoussi/AraT5_Darija_to_MSA"
+>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
+>>> model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Example translation
-input_text = "أه, يالاه رجاعت غير من شهار لعاسال ديالي ف شفشاون"
-inputs = tokenizer(input_text, return_tensors="pt", padding=True)
-translated = model.generate(**inputs)
-output_text = tokenizer.decode(translated[0], skip_special_tokens=True)
-print(output_text)
-# نعم، لقد عدت للتو من شهر العسل في شفشاون
+>>> input_text = "أه, يالاه رجاعت غير من شهار لعاسال ديالي ف شفشاون"
+>>> inputs = tokenizer(input_text, return_tensors="pt", padding=True)
+>>> translated = model.generate(**inputs)
+>>> output_text = tokenizer.decode(translated[0], skip_special_tokens=True)
+>>> print(output_text)
+```