prevent empty suggestion
Browse files
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 |
|