Update README.md
Browse files
README.md
CHANGED
|
@@ -14,6 +14,7 @@ datasets:
|
|
| 14 |
### Fine tuned MT5 base model with Sinhala Wikipedia Dataset (Experimentally continues training)
|
| 15 |
|
| 16 |
This model is fine tuned with articles from Sinhala Wikipedia for article generation. Used around 10,000 articles for training and fine tuned arround 100 times.
|
|
|
|
| 17 |
|
| 18 |
### How to use
|
| 19 |
|
|
@@ -44,4 +45,6 @@ tokenizer = AutoTokenizer.from_pretrained('google/mt5-base')
|
|
| 44 |
from transformers import pipeline
|
| 45 |
generator = pipeline(model='Suchinthana/MT5-Sinhala-Wikigen-Experimental', tokenizer=tokenizer)
|
| 46 |
generator("writeWiki: මානව ආහාර", do_sample=True, max_length=180)
|
| 47 |
-
```
|
|
|
|
|
|
|
|
|
| 14 |
### Fine tuned MT5 base model with Sinhala Wikipedia Dataset (Experimentally continues training)
|
| 15 |
|
| 16 |
This model is fine tuned with articles from Sinhala Wikipedia for article generation. Used around 10,000 articles for training and fine tuned arround 100 times.
|
| 17 |
+
|
| 18 |
|
| 19 |
### How to use
|
| 20 |
|
|
|
|
| 45 |
from transformers import pipeline
|
| 46 |
generator = pipeline(model='Suchinthana/MT5-Sinhala-Wikigen-Experimental', tokenizer=tokenizer)
|
| 47 |
generator("writeWiki: මානව ආහාර", do_sample=True, max_length=180)
|
| 48 |
+
```
|
| 49 |
+
|
| 50 |
+
If this model shows overfitting an early stopped version available at [non experimental repo.](https://huggingface.co/Suchinthana/MT-5-Sinhala-Wikigen)
|