captain-awesome commited on
Commit
ea857a1
·
verified ·
1 Parent(s): e1421e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -73,6 +73,7 @@ try:
73
  tools = mcp_client.get_tools()
74
 
75
  model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
 
76
 
77
  # model_id = "unsloth/Llama-3.2-1B"
78
  # model = AutoModelForCausalLM.from_pretrained(
 
73
  tools = mcp_client.get_tools()
74
 
75
  model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
76
+ agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
77
 
78
  # model_id = "unsloth/Llama-3.2-1B"
79
  # model = AutoModelForCausalLM.from_pretrained(