aiassociates's picture
Update README.md
4984dc7
|
raw
history blame
1 kB
metadata
language: de
tags:
  - grammar
  - text2text-generation
license: cc-by-nc-sa-4.0
datasets:
  - jfleg

T5 Grammar Correction

This model does restore upper and lower case as well as punctuation. It was trained with Happy Transformer on the german wikipedia dump.

Usage

pip install happytransformer

from happytransformer import HappyTextToText, TTSettings
happy_tt = HappyTextToText("T5", "aiassociates/t5-small-grammar-correction-german")
args = TTSettings(num_beams=5, min_length=1)
# Add the prefix "grammar: " before each input 
result = happy_tt.generate_text("grammar: hier ein kleines beispiel", args=args)
print(result.text) # Hier ein kleines Beispiel.

Authors

David Hustadt: dh@ai.associates

About us

AI.Associates

LinkedIn

We're always looking for developers to join us. Feel free to contact us careers@ai.associates