TedYeh commited on
Commit
4ac7ae8
1 Parent(s): 939c3ef

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ model = T5ForConditionalGeneration.from_pretrained("CodeTed/traditional_CSC_t5")
5
 
6
 
7
  def cged_correction(sentence = '為了降低少子化,政府可以堆動獎勵生育的政策。'):
8
- input_ids = tokenizer('糾正句子裡的錯字:' + sentence + '_輸出句:', return_tensors="pt").input_ids
9
- outputs = model.generate(input_ids, max_length=256)
10
  edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
  return edited_text
12
 
 
5
 
6
 
7
  def cged_correction(sentence = '為了降低少子化,政府可以堆動獎勵生育的政策。'):
8
+ input_ids = tokenizer('糾正句子裡的錯字:' + sentence, return_tensors="pt").input_ids
9
+ outputs = model.generate(input_ids, max_length=200)
10
  edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
  return edited_text
12