Luigi commited on
Commit
72f7051
·
1 Parent(s): d9961e2

prevent empty suggestion

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -40,7 +40,9 @@ def suggest_next(text, model_name, k, m):
40
  do_sample=False,
41
  early_stopping=True
42
  )
43
- suggestions = [out["generated_text"][len(text):] for out in outs]
 
 
44
  # 重置選值並更新選項
45
  return update(choices=suggestions, value=None)
46
 
 
40
  do_sample=False,
41
  early_stopping=True
42
  )
43
+ suggestions = [out["generated_text"][len(text):].strip() for out in outs]
44
+ suggestions = [s for s in suggestions if s]
45
+
46
  # 重置選值並更新選項
47
  return update(choices=suggestions, value=None)
48