rajaramesh commited on
Commit
43987fa
·
verified ·
1 Parent(s): 7e9c7b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  from smolagents import InferenceClientModel, CodeAgent#, ToolCollection
4
  from smolagents.mcp_client import MCPClient
5
 
 
6
 
7
  try:
8
  mcp_client = MCPClient(
@@ -14,7 +15,9 @@ try:
14
  tools = mcp_client.get_tools()
15
  print(f"my tools: {[*tools]}")
16
 
17
- model = InferenceClientModel()
 
 
18
  agent = CodeAgent(tools=[*tools], model=model)
19
 
20
  demo = gr.ChatInterface(
 
3
  from smolagents import InferenceClientModel, CodeAgent#, ToolCollection
4
  from smolagents.mcp_client import MCPClient
5
 
6
+ import os
7
 
8
  try:
9
  mcp_client = MCPClient(
 
15
  tools = mcp_client.get_tools()
16
  print(f"my tools: {[*tools]}")
17
 
18
+ model = InferenceClientModel(model_id="deepseek-ai/DeepSeek-R1",
19
+ provider="together",
20
+ token=os.getenv('HF_TOKEN'))
21
  agent = CodeAgent(tools=[*tools], model=model)
22
 
23
  demo = gr.ChatInterface(