Suchinthana commited on
Commit
baa557e
1 Parent(s): 4234a58

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -0
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
+ ```