ruanchaves commited on
Commit
8ac6b3b
1 Parent(s): bd9b3b2

feat: model settings

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -84,7 +84,11 @@ def parse_candidates(candidates):
84
 
85
  def predict(s1, language, candidates, use_reranker, topk, steps):
86
  hashtag_list = [s1]
87
- segmentation = model_dict.get(language, "english (fast)").segment(hashtag_list, use_reranker=use_reranker, return_ranks=True, topk=topk, steps=steps)
 
 
 
 
88
  segmenter_df = format_dataframe(segmentation.segmenter_rank)
89
  reranker_df = format_dataframe(segmentation.reranker_rank)
90
 
 
84
 
85
  def predict(s1, language, candidates, use_reranker, topk, steps):
86
  hashtag_list = [s1]
87
+ if language:
88
+ chosen_model = model_dict[language]
89
+ else:
90
+ chosen_model = model_dict["english (fast)"]
91
+ segmentation = chosen_model.segment(hashtag_list, use_reranker=use_reranker, return_ranks=True, topk=topk, steps=steps)
92
  segmenter_df = format_dataframe(segmentation.segmenter_rank)
93
  reranker_df = format_dataframe(segmentation.reranker_rank)
94