Commit
·
baa557e
1
Parent(s):
4234a58
Update README.md
Browse files
README.md
CHANGED
@@ -6,4 +6,22 @@ widget:
|
|
6 |
- text: 'writeWiki: Language Model'
|
7 |
language:
|
8 |
- en
|
|
|
|
|
9 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
- text: 'writeWiki: Language Model'
|
7 |
language:
|
8 |
- en
|
9 |
+
datasets:
|
10 |
+
- wikipedia
|
11 |
---
|
12 |
+
|
13 |
+
### Fine tuned T5 base model with Simple English Wikipedia Dataset
|
14 |
+
|
15 |
+
This model is fine tuned with articles from Simple English Wikipedia for article generation.
|
16 |
+
|
17 |
+
### How to use
|
18 |
+
|
19 |
+
We have to use **"writeWiki: "** part at the begining of each prompt. Used around 25,000 articles for training.
|
20 |
+
|
21 |
+
You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
|
22 |
+
|
23 |
+
```py
|
24 |
+
>>> from transformers import pipeline
|
25 |
+
>>> generator = pipeline('text2text-generation', model='Suchinthana/T5-Base-Wikigen')
|
26 |
+
>>> generator("writeWiki: Microcontroller", do_sample=True, max_length=250)
|
27 |
+
```
|