maytemuma commited on
Commit
ce7a508
·
verified ·
1 Parent(s): eec650c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -9,8 +9,10 @@ from smolagents import (
9
  VisitWebpageTool,
10
  InferenceClientModel,
11
  tool,
 
12
  )
13
 
 
14
  # --- Constants ---
15
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
 
@@ -251,11 +253,11 @@ class BasicAgent:
251
  print("Initializing SmolAgent for GAIA benchmark...")
252
 
253
  # Use default model -> smolagents auto-select the provider
254
- model = InferenceClientModel(
255
- token=os.getenv("HF_TOKEN"),
256
- timeout=180,
257
- max_tokens=2096,
258
  temperature=0.1,
 
259
  )
260
 
261
  self.agent = CodeAgent(
 
9
  VisitWebpageTool,
10
  InferenceClientModel,
11
  tool,
12
+ LiteLLMModel,
13
  )
14
 
15
+
16
  # --- Constants ---
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
 
 
253
  print("Initializing SmolAgent for GAIA benchmark...")
254
 
255
  # Use default model -> smolagents auto-select the provider
256
+ model = LiteLLMModel(
257
+ model_id="groq/llama-3.3-70b-versatile",
258
+ api_key=os.getenv("GROQ_API_KEY"),
 
259
  temperature=0.1,
260
+ max_tokens=2096,
261
  )
262
 
263
  self.agent = CodeAgent(