Update app.py

app.py

@@ -5,9 +5,9 @@ from cijiang.utils import print_results
 rhymer = CiJiangRhymer(strict=False, tone=True)
 
 def get_results(input_text, target_rhyme, beam_width, num_candidates, top_results):
-    out = rhymer.get_rhymes(text=input_text, target_rhyme=target_rhyme, beam_width=beam_width, num_candidates=num_candidates)
-    mask_count = text.count("[M]")
-    context = text.split('[M]')[0]
+    out = rhymer.get_rhymes(text_with_placeholder=input_text, target_rhyme=target_rhyme, beam_width=beam_width, num_candidates=num_candidates)
+    mask_count = input_text.count("[M]")
+    context = input_text.split('[M]')[0]
 
     all_outputs = f"======= 韵脚: |{target_rhyme}|\n"
     for i, (seq, log_prob) in enumerate(out[:top_results]):