lingyu98 commited on
Commit
862d92c
·
verified ·
1 Parent(s): a6d1e76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from cijiang.utils import print_results
4
 
5
  rhymer = CiJiangRhymer(strict=False, tone=True)
6
 
7
- def get_results(input_text, target_rhyme, beam_width=20, num_candidates=4000):
8
  out = rhymer.get_rhymes(text=input_text, target_rhyme=target_rhyme, beam_width=beam_width, num_candidates=num_candidates)
9
  mask_count = text.count("[M]")
10
  context = text.split('[M]')[0]
 
4
 
5
  rhymer = CiJiangRhymer(strict=False, tone=True)
6
 
7
+ def get_results(input_text, target_rhyme, beam_width, num_candidates, top_results):
8
  out = rhymer.get_rhymes(text=input_text, target_rhyme=target_rhyme, beam_width=beam_width, num_candidates=num_candidates)
9
  mask_count = text.count("[M]")
10
  context = text.split('[M]')[0]