fix: fix function bard_generate_stream's output

fix function bard_generate_stream's output
This commit is contained in:
xuyuan23 2023-07-27 11:11:59 +08:00
parent 572c336b81
commit c4638bfde3
2 changed files with 3 additions and 4 deletions

View File

@ -34,6 +34,5 @@ def bard_generate_stream(model, tokenizer, params, device, context_len=2048):
response = bardapi.core.Bard(token).get_answer(last_user_input["content"])
if response is not None and response.get("content") is not None:
yield str(response["content"])
yield f"bard response error: {str(response)}"
else:
yield f"bard response error: {str(response)}"

View File

@ -1,3 +1,3 @@
"""
a tool to discovery sensitive data
"""
"""