ashishkat commited on
Commit
913316e
·
1 Parent(s): 82d79a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -1,16 +1,22 @@
1
- """
 
2
  tokenized_text = tokenizer.encode_plus(
3
  str(text_input),
4
  return_attention_mask = True,
5
  return_tensors="pt")
6
 
 
7
  generated_token = model.generate(
8
  input_ids = tokenized_text["input_ids"],
9
  attention_mask=tokenized_text["attention_mask"],
10
  max_length = 256,
11
  use_cache=True,
12
  )
 
 
13
  summarized_paragraph = [
14
  tokenizer.decode(token_ids=ids, skip_special_tokens=True) for ids in generated_token
15
  ]
16
- """
 
 
 
1
+ ```
2
+ ### encode the input paragraph to summarize
3
  tokenized_text = tokenizer.encode_plus(
4
  str(text_input),
5
  return_attention_mask = True,
6
  return_tensors="pt")
7
 
8
+ ### pass the tokenized paragraph in model
9
  generated_token = model.generate(
10
  input_ids = tokenized_text["input_ids"],
11
  attention_mask=tokenized_text["attention_mask"],
12
  max_length = 256,
13
  use_cache=True,
14
  )
15
+
16
+ ### decode the summarized paragraph token
17
  summarized_paragraph = [
18
  tokenizer.decode(token_ids=ids, skip_special_tokens=True) for ids in generated_token
19
  ]
20
+
21
+ Summarized paragraph is the final result.
22
+ ```