yetessam commited on
Commit
624116f
·
verified ·
1 Parent(s): 62c62fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, HfApiModel
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 = HfApiModel(
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,