Luigi commited on
Commit
9d21dc2
·
1 Parent(s): 83bd284

replace stript with rstript (ie. prediciton can insert prefixeing space )

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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):].strip()
71
  if not snippet:
72
  continue
73
- converted = cc.convert(snippet).strip()
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