Hugging Face's logo

Spaces:
ruanchaves
/
hashtag-segmentation
Runtime error

App Files Community
ruanchaves commited on
Commit
bd9b3b2
•
1 Parent(s): e66dfc8

feat: model settings

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def parse_candidates(candidates):
84
 
85
  def predict(s1, language, candidates, use_reranker, topk, steps):
86
  hashtag_list = [s1]
87
- segmentation = model_dict[language].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
+ 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