Update README.md

README.md

@@ -14,13 +14,12 @@ This model is fine-tuned to generate formal text from informal text based on the
 
 ```python
 
-from transformers import  AutoModelForSeq2SeqLM, MT5Tokenizer
+from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
+import torch
 
 model = T5ForConditionalGeneration.from_pretrained('PardisSzah/PersianTextFormalizer')
 tokenizer = AutoTokenizer.from_pretrained('PardisSzah/PersianTextFormalizer')
 
-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')