Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ The only limitation you might face is, to get the best results, you will have to
|
|
| 40 |
|
| 41 |
Use the code below to get started with the model.
|
| 42 |
|
|
|
|
| 43 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 44 |
import torch
|
| 45 |
|
|
@@ -63,6 +64,8 @@ summary = tokenizer.batch_decode(sequences, skip_special_tokens=True) # Decode
|
|
| 63 |
## Print the generated summary
|
| 64 |
print(summary)
|
| 65 |
|
|
|
|
|
|
|
| 66 |
## Feel free to play around with the hyperparameters in the generate, or some other parameters to include for experimentation purpose.
|
| 67 |
|
| 68 |
|
|
|
|
| 40 |
|
| 41 |
Use the code below to get started with the model.
|
| 42 |
|
| 43 |
+
---
|
| 44 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 45 |
import torch
|
| 46 |
|
|
|
|
| 64 |
## Print the generated summary
|
| 65 |
print(summary)
|
| 66 |
|
| 67 |
+
---
|
| 68 |
+
|
| 69 |
## Feel free to play around with the hyperparameters in the generate, or some other parameters to include for experimentation purpose.
|
| 70 |
|
| 71 |
|