ruanchaves commited on
Commit
87ced4a
·
1 Parent(s): 8ac6b3b

feat: model settings

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -88,6 +88,10 @@ def predict(s1, language, candidates, use_reranker, topk, steps):
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)
 
88
  chosen_model = model_dict[language]
89
  else:
90
  chosen_model = model_dict["english (fast)"]
91
+
92
+ if not all([topk, steps]):
93
+ return None, None
94
+
95
  segmentation = chosen_model.segment(hashtag_list, use_reranker=use_reranker, return_ranks=True, topk=topk, steps=steps)
96
  segmenter_df = format_dataframe(segmentation.segmenter_rank)
97
  reranker_df = format_dataframe(segmentation.reranker_rank)