Hugging Face's logo

Spaces:
markqiu
/
prinvest_mate
Sleeping

App Files Community
chinwh2019 commited on
Commit
d3b93fb
•
1 Parent(s): 98d8027

fix: update chat_func fix int issue (#574)

Browse files
Files changed (1) hide show
  1. modules/chat_func.py +1 -1
modules/chat_func.py CHANGED
@@ -244,7 +244,7 @@ def predict_all(
244
  all_token_counts[-1] += count_token(construct_assistant(content))
245
  else:
246
  all_token_counts[-1] = total_token_count - sum(all_token_counts)
247
- status_text = construct_token_message(total_token_count)
248
  return chatbot, history, status_text, all_token_counts
249
  except KeyError:
250
  status_text = standard_error_msg + str(response)
 
244
  all_token_counts[-1] += count_token(construct_assistant(content))
245
  else:
246
  all_token_counts[-1] = total_token_count - sum(all_token_counts)
247
+ status_text = construct_token_message([total_token_count])
248
  return chatbot, history, status_text, all_token_counts
249
  except KeyError:
250
  status_text = standard_error_msg + str(response)