Update README.md

README.md

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