Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ class BasicAgent:
|
|
40 |
# )
|
41 |
|
42 |
self.model = TransformersModel(model_id="HuggingFaceTB/SmolLM2-1.7B-Instruct", device_map="auto", max_new_tokens=1000)
|
43 |
-
self.agent = ToolCallingAgent(tools=[search_tool], model=model, verbosity_level=2)
|
44 |
|
45 |
print("Agent initialized successfully.")
|
46 |
|
|
|
40 |
# )
|
41 |
|
42 |
self.model = TransformersModel(model_id="HuggingFaceTB/SmolLM2-1.7B-Instruct", device_map="auto", max_new_tokens=1000)
|
43 |
+
self.agent = ToolCallingAgent(tools=[search_tool], model=self.model, verbosity_level=2)
|
44 |
|
45 |
print("Agent initialized successfully.")
|
46 |
|