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

在線預測

你還可以在託管推理API下輸入文本，在線獲取預測結果。