captain-awesome commited on
Commit
f213c9b
·
verified ·
1 Parent(s): 9b51b5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from mcp.client.stdio import StdioServerParameters
4
  from smolagents import InferenceClientModel, CodeAgent
5
  from smolagents.mcp_client import MCPClient
6
  from transformers import pipeline
 
7
 
8
  # Initialize the MCP client correctly
9
  try:
 
4
  from smolagents import InferenceClientModel, CodeAgent
5
  from smolagents.mcp_client import MCPClient
6
  from transformers import pipeline
7
+ from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
  # Initialize the MCP client correctly
10
  try: