Update app.py
Browse files
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(
|