Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import pytz # Had to give it permission in Code agent
|
|
| 10 |
from ui.contentagentui import ContentAgentUI
|
| 11 |
# from ui.gradioui import GradioUI
|
| 12 |
|
| 13 |
-
from smolagents import CodeAgent,
|
| 14 |
from status_check import resolve_endpoint, is_endpoint_healthy
|
| 15 |
|
| 16 |
from tools.tools import load_tools
|
|
@@ -24,7 +24,7 @@ def initialize_agent(endpoint_uri: str):
|
|
| 24 |
combined_tools = load_tools() ## from the tools folder (don't remove final answer)
|
| 25 |
|
| 26 |
"""Initialize and configure the CodeAgent"""
|
| 27 |
-
model =
|
| 28 |
max_tokens=2096,
|
| 29 |
temperature=0.5,
|
| 30 |
model_id=endpoint_uri,
|
|
|
|
| 10 |
from ui.contentagentui import ContentAgentUI
|
| 11 |
# from ui.gradioui import GradioUI
|
| 12 |
|
| 13 |
+
from smolagents import CodeAgent, InferenceClientModel
|
| 14 |
from status_check import resolve_endpoint, is_endpoint_healthy
|
| 15 |
|
| 16 |
from tools.tools import load_tools
|
|
|
|
| 24 |
combined_tools = load_tools() ## from the tools folder (don't remove final answer)
|
| 25 |
|
| 26 |
"""Initialize and configure the CodeAgent"""
|
| 27 |
+
model = InferenceClientModel (
|
| 28 |
max_tokens=2096,
|
| 29 |
temperature=0.5,
|
| 30 |
model_id=endpoint_uri,
|