chat_bot.py

import yaml
from smolagents import (
    load_tool,
    CodeAgent,
    HfApiModel,
    GradioUI,
    VisitWebpageTool,
)

model_id='Qwen/Qwen2.5-Coder-32B-Instruct'

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id=model_id,
custom_role_conversions=None,
)


# Import tool from Hub
image_generation_tool = load_tool("m-ric/text-to-image", trust_remote_code=True)

model = HfApiModel(model_id)

visitWebpageTool = VisitWebpageTool()

agent = CodeAgent(tools=[visitWebpageTool, image_generation_tool], model=model)

gr = GradioUI(agent).launch()