Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
| 255 |
-
|
| 256 |
-
|
| 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(
|