jcleee commited on
Commit
1867ee3
·
verified ·
1 Parent(s): a8e10cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -70,8 +70,6 @@ with open("agent.json", "r") as f:
70
  model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
73
- from smolagents import HfApiModel
74
-
75
  model = LiteLLMModel(
76
  model_id="gemini/gemini-2.0-flash-lite",
77
  api_key=os.getenv("GEMINI_API_KEY"),
@@ -79,9 +77,6 @@ model = LiteLLMModel(
79
  max_tokens=1024,
80
  )
81
 
82
- model.last_input_token_count = 0
83
- model.last_output_token_count = 0
84
-
85
 
86
 
87
  # === IMPORT TOOL FROM HUB ===
 
70
  model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
 
 
73
  model = LiteLLMModel(
74
  model_id="gemini/gemini-2.0-flash-lite",
75
  api_key=os.getenv("GEMINI_API_KEY"),
 
77
  max_tokens=1024,
78
  )
79
 
 
 
 
80
 
81
 
82
  # === IMPORT TOOL FROM HUB ===