rajeshthangaraj1 commited on
Commit
7802f3a
·
verified ·
1 Parent(s): 94eb831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,6 +5,7 @@ from smolagents import ToolCollection, CodeAgent
5
  from smolagents import CodeAgent, InferenceClientModel
6
  from smolagents.mcp_client import MCPClient
7
 
 
8
 
9
  try:
10
  mcp_client = MCPClient(
@@ -13,7 +14,7 @@ try:
13
  )
14
  tools = mcp_client.get_tools()
15
 
16
- model = InferenceClientModel()
17
  agent = CodeAgent(tools=[*tools], model=model)
18
 
19
  def call_agent(message, history):
 
5
  from smolagents import CodeAgent, InferenceClientModel
6
  from smolagents.mcp_client import MCPClient
7
 
8
+ token = os.environ.get("HF_TOKEN")
9
 
10
  try:
11
  mcp_client = MCPClient(
 
14
  )
15
  tools = mcp_client.get_tools()
16
 
17
+ model = InferenceClientModel(token=token)
18
  agent = CodeAgent(tools=[*tools], model=model)
19
 
20
  def call_agent(message, history):