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