Update app.py

app.py

@@ -69,17 +69,20 @@ with open("agent.json", "r") as f:
 
 model_config = agent_config["model"]["data"]
 
-# === BUILD MODEL MANUALLY (SAFE FALLBACK) ===
+# === BUILD MODEL ===
 model = HfApiModel(
     model_id=model_config["model_id"],
-    token=os.getenv("HF_TOKEN"),  # Injected securely from Space secret
+    token=os.getenv("HF_TOKEN"),
     temperature=model_config.get("temperature", 0.5),
     max_tokens=model_config.get("max_tokens", 2048),
-    last_input_token_count=model_config.get("last_input_token_count", 0),
-    last_output_token_count=model_config.get("last_output_token_count", 0),
     custom_role_conversions=model_config.get("custom_role_conversions", None),
 )
 
+# ✅ Force these to be correct after init
+model.last_input_token_count = 0
+model.last_output_token_count = 0
+
+
 # === IMPORT TOOL FROM HUB ===
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)