Update app.py

app.py

@@ -72,5 +72,11 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+if agent.model.last_input_token_count is not None:
+    total_input_tokens += agent.model.last_input_token_count
+else:
+    # Handle the case where last_input_token_count is None
+    # For example, you can set total_input_tokens to a default value
+    total_input_tokens = 0
 
 GradioUI(agent).launch()