T5 Base Spellchecker

🚀 T5ベースのTransformerを使用したスペルチェッカー

このプロジェクトは、T5-Base Transformerを用いて構築されたシンプルなスペルチェッカーです。文章のスペルミスを自動的に修正することができます。

🚀 クイックスタート

以下の手順で、このスペルチェッカーを使用することができます。

基本的な使用法

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("Bhuvana/t5-base-spellchecker")

model = AutoModelForSeq2SeqLM.from_pretrained("Bhuvana/t5-base-spellchecker")


def correct(inputs):
    input_ids = tokenizer.encode(inputs,return_tensors='pt')
    sample_output = model.generate(
        input_ids,
        do_sample=True,
        max_length=50,
        top_p=0.99,
        num_return_sequences=1
    )
    res = tokenizer.decode(sample_output[0], skip_special_tokens=True)
    return res


text = "christmas is celbrated on decembr 25 evry ear"

print(correct(text))

このコードを実行すると、修正された文章が出力されます。

christmas is celebrated on december 25 every year

また、Hosted inference API上でテキストを入力することで、オンラインで予測結果を得ることもできます。