Update app.py

app.py

@@ -13,7 +13,7 @@ def get_results(input_text, target_rhyme, beam_width, num_candidates, top_result
     all_outputs = f"======= 韵脚: |{target_rhyme}|\n"
     for i, (seq, log_prob) in enumerate(out[:top_results]): 
         rhymes = seq[-mask_count:].split()
-        colored_rhymes = [Fore.RED + part + Style.RESET_ALL if idx < mask_count else part for idx, part in enumerate(rhymes)]
+        colored_rhymes = [part if idx < mask_count else part for idx, part in enumerate(rhymes)]
         colored_rhymes = ''.join(colored_rhymes)  # Join the parts back together
 
         all_outputs += f"{i+1}. {context}{colored_rhymes} (score: {log_prob:.3f})\n"
@@ -58,7 +58,7 @@ demo = gr.Interface(
         ),
         gr.Slider(
             minimum=1,
-            maximum=50,
+            maximum=100,
             value=20,
             step=1,
             label="Beam Width"
@@ -72,8 +72,8 @@ demo = gr.Interface(
         ),
         gr.Slider(
             minimum=1,
-            maximum=20,
-            value=10,
+            maximum=50,
+            value=20,
             step=1,
             label="Top Results to Show"
         )
@@ -88,12 +88,9 @@ demo = gr.Interface(
     examples=[
         ["[M][M][M][M]", "乱比高低", 20, 4000, 10],
         ["[M][M][M][M]", "luan4 bi3 gao1 di1", 20, 4000, 10],
-        ["[M][M][M]", "收视率", 20, 4000, 10],
         ["我总是天亮了[M][M]", "排队", 20, 4000, 10],
         ["我们两个人的verse价值一辆[M][M][M][M]", "韩国低级", 20, 4000, 10],
-        ["报名嘻[M][M]", "八个", 20, 4000, 10],
-        ["我还是觉得音乐的[M][M][M][M][M]", "背靠背钻研", 20, 4000, 10],
-        ["我还是觉得音乐的[M][M][M][M][M]", "背靠背钻研", 300, 4000, 10]
+        ["我还是觉得音乐的[M][M][M][M][M]", "背靠背钻研", 100, 4000, 10],
     ]
 )
 demo.launch()