Views
No views yet
pip install happytransformer 1from happytransformer import HappyTextToText, TTSettings
2
3happy_tt = HappyTextToText("T5", "fdemelo/t5-base-spell-correction-fr")
4
5args = TTSettings(num_beams=5, min_length=1)
6
7# Add the prefix "grammaire: " before each input
8result = happy_tt.generate_text("grammaire: Le vehicule a tombe encontrebas", args=args)
9
10print(result.text) # corrected sentence
11
12