Update app.py
Browse files
app.py
CHANGED
@@ -78,8 +78,8 @@ class BasicAgent:
|
|
78 |
|
79 |
# Initialize the agent
|
80 |
pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
|
81 |
-
self.
|
82 |
-
self.agent = CodeAgent(
|
83 |
|
84 |
# self.agent = CodeAgent(
|
85 |
# model=model,
|
|
|
78 |
|
79 |
# Initialize the agent
|
80 |
pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
|
81 |
+
self.model = LocalLLM(pipe)
|
82 |
+
self.agent = CodeAgent(model=self.model, tools=[], add_base_tools=True)
|
83 |
|
84 |
# self.agent = CodeAgent(
|
85 |
# model=model,
|