Spaces:
Running
on
Zero
Running
on
Zero
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 |
|