Update app.py
#18
by
sdeepanraj
- opened
app.py
CHANGED
@@ -30,12 +30,12 @@ class BasicAgent:
|
|
30 |
# custom_role_conversions=None,
|
31 |
#)
|
32 |
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
self.agent = CodeAgent(
|
38 |
-
model =
|
39 |
tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
|
40 |
#additional_authorized_imports = ["pandas", "numpy"]
|
41 |
)
|
|
|
30 |
# custom_role_conversions=None,
|
31 |
#)
|
32 |
|
33 |
+
model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-0.5B-Instruct",
|
34 |
+
max_tokens=2096,
|
35 |
+
token=token,
|
36 |
+
custom_role_conversions=None)
|
37 |
self.agent = CodeAgent(
|
38 |
+
model = model,
|
39 |
tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
|
40 |
#additional_authorized_imports = ["pandas", "numpy"]
|
41 |
)
|