Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import inspect
|
|
5 |
import pandas as pd
|
6 |
|
7 |
try:
|
8 |
-
from smolagents import CodeAgent, DuckDuckGoSearchTool,
|
9 |
SMOLAGENTS_AVAILABLE = True
|
10 |
except ImportError:
|
11 |
print("Warning: smolagents not available, using fallback implementation")
|
@@ -44,12 +44,7 @@ class BasicAgent:
|
|
44 |
if SMOLAGENTS_AVAILABLE:
|
45 |
try:
|
46 |
# Initialize the model (from Unit 2)
|
47 |
-
self.model =
|
48 |
-
max_tokens=2096,
|
49 |
-
temperature=0.5,
|
50 |
-
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
|
51 |
-
custom_role_conversions=None,
|
52 |
-
)
|
53 |
|
54 |
# Create CodeAgent with DuckDuckGoSearchTool (core Unit 2 concept)
|
55 |
self.agent = CodeAgent(
|
|
|
5 |
import pandas as pd
|
6 |
|
7 |
try:
|
8 |
+
from smolagents import CodeAgent, DuckDuckGoSearchTool, TransformersModel
|
9 |
SMOLAGENTS_AVAILABLE = True
|
10 |
except ImportError:
|
11 |
print("Warning: smolagents not available, using fallback implementation")
|
|
|
44 |
if SMOLAGENTS_AVAILABLE:
|
45 |
try:
|
46 |
# Initialize the model (from Unit 2)
|
47 |
+
self.model = TransformersModel(model_id="google/gemma-3-4b-it", torch_dtype=torch.float16)
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
# Create CodeAgent with DuckDuckGoSearchTool (core Unit 2 concept)
|
50 |
self.agent = CodeAgent(
|