Spaces:

markqiu
/

prinvest_mate

Sleeping

chinwh2019 commited on Apr 8, 2023

Commit

d3b93fb

unverified ·

1 Parent(s): 98d8027

fix: update chat_func fix int issue (#574)

Files changed (1) hide show

modules/chat_func.py CHANGED Viewed

@@ -244,7 +244,7 @@ def predict_all(
             all_token_counts[-1] += count_token(construct_assistant(content))
         else:
             all_token_counts[-1] = total_token_count - sum(all_token_counts)
-        status_text = construct_token_message(total_token_count)
         return chatbot, history, status_text, all_token_counts
     except KeyError:
         status_text = standard_error_msg + str(response)

             all_token_counts[-1] += count_token(construct_assistant(content))
         else:
             all_token_counts[-1] = total_token_count - sum(all_token_counts)
+        status_text = construct_token_message([total_token_count])
         return chatbot, history, status_text, all_token_counts
     except KeyError:
         status_text = standard_error_msg + str(response)