shri07 commited on
Commit
b3eea3c
1 Parent(s): 4c754b2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -1,3 +1,4 @@
 
1
  import transformers
2
  import datasets
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -12,4 +13,4 @@ input_text = '' #your input text here must be a string
12
  input = tokenize(input_text, return_tensors='pt', padding=True).to(device)
13
  model = model.to(device)
14
  output = model.generate(*input, max_length=1024)
15
- prediction = tokenize.decode(output[0],skip_special_tokens=True)
 
1
+ ```
2
  import transformers
3
  import datasets
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
13
  input = tokenize(input_text, return_tensors='pt', padding=True).to(device)
14
  model = model.to(device)
15
  output = model.generate(*input, max_length=1024)
16
+ prediction = tokenize.decode(output[0],skip_special_tokens=True)```