Update app.py
#20
by
sdeepanraj
- opened
app.py
CHANGED
@@ -30,13 +30,13 @@ class BasicAgent:
|
|
30 |
# custom_role_conversions=None,
|
31 |
#)
|
32 |
|
33 |
-
model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct",
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
self.agent = CodeAgent(
|
38 |
-
model =
|
39 |
-
tools = [DuckDuckGoSearchTool()
|
40 |
#additional_authorized_imports = ["pandas", "numpy"]
|
41 |
)
|
42 |
SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
|
|
|
30 |
# custom_role_conversions=None,
|
31 |
#)
|
32 |
|
33 |
+
#model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct",
|
34 |
+
# max_tokens=2096,
|
35 |
+
# token=token,
|
36 |
+
# custom_role_conversions=None)
|
37 |
self.agent = CodeAgent(
|
38 |
+
model = HfApiModel(),
|
39 |
+
tools = [DuckDuckGoSearchTool()],
|
40 |
#additional_authorized_imports = ["pandas", "numpy"]
|
41 |
)
|
42 |
SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
|