Update app.py

app.py

@@ -9,9 +9,10 @@ def cged_correction(sentence, function):
     prompt = {"錯別字校正":"糾正句子中的錯字：", "文法校正":"糾正句子中的錯誤：", 
     "文本重構":"在不改動文意的情況下改寫句子：", "文本簡化":"在不改動文意的情況下改寫句子：", "整體校正":"修改句子的錯誤或使其更通順："}
     #input_ids = tokenizer(prompt[function] + sentence, return_tensors="pt").input_ids
-    outputs = model.predict([prompt[function] + sentence + "_輸出句："])
+    for _ in range(3):
+        sentence = model.predict([prompt[function] + sentence + "_輸出句："])[0]
     #edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return outputs[0]
+    return sentence
 
 with gr.Blocks() as demo:
     gr.Markdown(