replace stript with rstript (ie. prediciton can insert prefixeing space )
Browse files
app.py
CHANGED
|
@@ -67,10 +67,10 @@ def suggest_next(text, model_name, k, m, num_beam_groups, diversity_penalty):
|
|
| 67 |
# 提取純下文、過濾空字串、繁體化、確保 strip 處理
|
| 68 |
suggestions = set()
|
| 69 |
for out in outs:
|
| 70 |
-
snippet = out["generated_text"][len(text):].
|
| 71 |
if not snippet:
|
| 72 |
continue
|
| 73 |
-
converted = cc.convert(snippet)
|
| 74 |
suggestions.add(converted)
|
| 75 |
suggestions = list(suggestions)
|
| 76 |
|
|
|
|
| 67 |
# 提取純下文、過濾空字串、繁體化、確保 strip 處理
|
| 68 |
suggestions = set()
|
| 69 |
for out in outs:
|
| 70 |
+
snippet = out["generated_text"][len(text):].rstrip()
|
| 71 |
if not snippet:
|
| 72 |
continue
|
| 73 |
+
converted = cc.convert(snippet)
|
| 74 |
suggestions.add(converted)
|
| 75 |
suggestions = list(suggestions)
|
| 76 |
|