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