Spaces:
Runtime error
Runtime error
Update chat_func.py
Browse files- chat_func.py +1 -1
chat_func.py
CHANGED
|
@@ -195,7 +195,7 @@ def stream_predict(
|
|
| 195 |
chatbot[-1] = (chatbot[-1][0], partial_words+display_append)
|
| 196 |
all_token_counts[-1] += 1
|
| 197 |
yield get_return_value()
|
| 198 |
-
|
| 199 |
|
| 200 |
def main():
|
| 201 |
# 调用stream_predict函数获取结果...
|
|
|
|
| 195 |
chatbot[-1] = (chatbot[-1][0], partial_words+display_append)
|
| 196 |
all_token_counts[-1] += 1
|
| 197 |
yield get_return_value()
|
| 198 |
+
return all_token_counts
|
| 199 |
|
| 200 |
def main():
|
| 201 |
# 调用stream_predict函数获取结果...
|