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(
|