Lu032145 commited on
Commit
2b64ec8
·
verified ·
1 Parent(s): dc9b88f

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +1 -1
Gradio_UI.py CHANGED
@@ -144,7 +144,7 @@ def stream_to_gradio(
144
  if hasattr(agent.model, "last_input_token_count"):
145
  if agent.model.last_input_token_count is not None:
146
  total_output_tokens += agent.model.last_output_token_count
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
 
 
144
  if hasattr(agent.model, "last_input_token_count"):
145
  if agent.model.last_input_token_count is not None:
146
  total_output_tokens += agent.model.last_output_token_count
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