Hugging Face's logo

Spaces:
Baladuri
/
First_agent_template
Sleeping

App Files Community
Baladuri commited on
Commit
4e5db54
·
verified ·
1 Parent(s): 0045560

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +2 -2
Gradio_UI.py CHANGED
@@ -145,8 +145,8 @@ def stream_to_gradio(
145
  total_input_tokens += agent.model.last_input_token_count or 0
146
  total_output_tokens += agent.model.last_output_token_count or 0
147
  if isinstance(step_log, ActionStep):
148
- step_log.input_token_count = agent.model.last_input_token_count
149
- step_log.output_token_count = agent.model.last_output_token_count
150
 
151
  for message in pull_messages_from_step(
152
  step_log,
 
145
  total_input_tokens += agent.model.last_input_token_count or 0
146
  total_output_tokens += agent.model.last_output_token_count or 0
147
  if isinstance(step_log, ActionStep):
148
+ step_log.input_token_count = agent.model.last_input_token_count or 0
149
+ step_log.output_token_count = agent.model.last_output_token_count or 0
150
 
151
  for message in pull_messages_from_step(
152
  step_log,