Update README.md

README.md

@@ -13,13 +13,12 @@ This model is fine-tuned to generate informal text from formal text based on the
 
 ```python
 
-from transformers import  AutoModelForSeq2SeqLM, MT5Tokenizer
-
+from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
+import torch
 model = T5ForConditionalGeneration.from_pretrained('PardisSzah/BasePersianEase')
 tokenizer = AutoTokenizer.from_pretrained('PardisSzah/BasePersianEase')
 
-from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
-import torch
+
 pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)
 def test_model(text):
   device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')