PardisSzah commited on
Commit
974a3fa
1 Parent(s): 8e5f06c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -13,13 +13,12 @@ This model is fine-tuned to generate informal text from formal text based on the
13
 
14
  ```python
15
 
16
- from transformers import AutoModelForSeq2SeqLM, MT5Tokenizer
 
17
 
18
  model = T5ForConditionalGeneration.from_pretrained('PardisSzah/PersianEase')
19
  tokenizer = AutoTokenizer.from_pretrained('PardisSzah/PersianEase')
20
 
21
- from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
22
- import torch
23
  pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)
24
  def test_model(text):
25
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
13
 
14
  ```python
15
 
16
+ from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
17
+ import torch
18
 
19
  model = T5ForConditionalGeneration.from_pretrained('PardisSzah/PersianEase')
20
  tokenizer = AutoTokenizer.from_pretrained('PardisSzah/PersianEase')
21
 
 
 
22
  pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)
23
  def test_model(text):
24
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')