Spaces:

CodeTed
/

chinese_spelling_error_correction

Running

TedYeh commited on Nov 15, 2023

Commit

c2ec08b

1 Parent(s): 772bcbe

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,9 @@ from transformers import AutoTokenizer, T5ForConditionalGeneration
 model = T5Model('t5', "CodeTed/traditional_CSC_t5", args={"eval_batch_size": 1}, cuda_device=-1, evaluate=True)
 def cged_correction(sentence = '為了降低少子化，政府可以堆動獎勵生育的政策。'):
-    outputs = model.predict(["糾正句子中的錯字：" + sentence + "_輸出句："])
     return outputs[0]
 with gr.Blocks() as demo:

 model = T5Model('t5', "CodeTed/traditional_CSC_t5", args={"eval_batch_size": 1}, cuda_device=-1, evaluate=True)
 def cged_correction(sentence = '為了降低少子化，政府可以堆動獎勵生育的政策。'):
+    for _ in range(3):
+        outputs = model.predict(["糾正句子中的錯字：" + sentence + "_輸出句："])
+        sentence = outputs[0]
     return outputs[0]
 with gr.Blocks() as demo: