Spaces:

CodeTed
/

chinese_spelling_error_correction

Running

TedYeh commited on Nov 15, 2023

Commit

03bdda8

1 Parent(s): 64fe5a1

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ model = T5ForConditionalGeneration.from_pretrained("CodeTed/traditional_CSC_t5")
 def cged_correction(sentence = '為了降低少子化，政府可以堆動獎勵生育的政策。'):
-    input_ids = tokenizer('糾正句子裡的錯字: ' + sentence, return_tensors="pt").input_ids
     outputs = model.generate(input_ids, max_length=256)
     edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return edited_text

 def cged_correction(sentence = '為了降低少子化，政府可以堆動獎勵生育的政策。'):
+    input_ids = tokenizer('糾正句子裡的錯字：' + sentence + "_輸出句：", return_tensors="pt").input_ids
     outputs = model.generate(input_ids, max_length=256)
     edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return edited_text