Update app.py

app.py

@@ -4,9 +4,10 @@ from cijiang.utils import print_results
 from colorama import Fore, Style, init
 import spaces
 
-@spaces.GPU  # Add this decorator
+
 rhymer = CiJiangRhymer(strict=False, tone=True)
 
+@spaces.GPU  # Add this decorator
 def get_results(input_text, target_rhyme, beam_width, num_candidates, top_results):
     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]")
@@ -22,28 +23,7 @@ def get_results(input_text, target_rhyme, beam_width, num_candidates, top_result
     all_outputs += "=" * 40
     return all_outputs
 
-# rhymer = CiJiangRhymer(strict=False, tone=True)
-
-# # 使用韵脚的汉字押韵
-# print_results(rhymer, text="[M][M][M][M]", target_rhyme="乱比高低")
-
-# # 使用韵脚的拼音押韵
-# print_results(rhymer, text="[M][M][M][M]", target_rhyme="luan4 bi3 gao1 di1")
-
-# # 允许声调不同
-# rhymer.tone = False
-# print_results(rhymer, text="[M][M][M]", target_rhyme="收视率")
-# rhymer.tone = True
-
-# # 上下文押韵
-# print_results(rhymer, text="我总是天亮了[M][M]", target_rhyme="排队")
-# print_results(rhymer, text="我们两个人的verse价值一辆[M][M][M][M]", target_rhyme="韩国低级")
 
-# # # 上下文非词汇押韵
-# print_results(rhymer, text="报名嘻[M][M]", target_rhyme="八个")
-# # 长韵脚需要更广的搜索空间
-# print_results(rhymer, text="我还是觉得音乐的[M][M][M][M][M]", target_rhyme="背靠背钻研")
-# print_results(rhymer, text="我还是觉得音乐的[M][M][M][M][M]", target_rhyme="背靠背钻研", beam_width=300)
 demo = gr.Interface(
     fn=get_results,
     inputs=[