Spaces:
Runtime error
Runtime error
Commit
·
87ced4a
1
Parent(s):
8ac6b3b
feat: model settings
Browse files
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)
|