Luigi commited on
Commit
4a676fe
·
1 Parent(s): 28bd11e

remove extra cleanning on suggestion list

Browse files
Files changed (1) hide show
  1. app.py +4 -26
app.py CHANGED
@@ -65,38 +65,16 @@ def suggest_next(text, model_name, k, m, num_beam_groups, diversity_penalty):
65
  outs = gen_pipe(text, **gen_kwargs)
66
 
67
  # 提取純下文、過濾空字串、繁體化、確保 strip 處理
68
- raw_suggestions = []
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
- raw_suggestions.append(converted)
 
75
 
76
- # 去重 (基於 strip 後內容)
77
- unique_suggestions = []
78
- seen = set()
79
- for s in raw_suggestions:
80
- key = s
81
- if key not in seen:
82
- seen.add(key)
83
- unique_suggestions.append(key)
84
-
85
- # 合併共同前綴
86
- cprint(f'unique_suggestions: {unique_suggestions}','blue')
87
- merged_prefixes = unique_suggestions
88
- cprint(f"merged_prefixes: {merged_prefixes}",'red')
89
-
90
- # 最終去重並移除空項 (基於 strip 後內容)
91
- final_suggestions = []
92
- seen_final = set()
93
- for s in merged_prefixes:
94
- key = s.strip()
95
- if key and key not in seen_final:
96
- seen_final.add(key)
97
- final_suggestions.append(key)
98
-
99
- return update(choices=final_suggestions, value=None)
100
 
101
 
102
  def append_suggestion(current, choice):
 
65
  outs = gen_pipe(text, **gen_kwargs)
66
 
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
 
77
+ return update(choices=suggestions, value=None)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
78
 
79
 
80
  def append_suggestion(current, choice):