runtime error

Exit code: 1. Reason: tool.py: 0%| | 0.00/635 [00:00<?, ?B/s] tool.py: 100%|██████████| 635/635 [00:00<00:00, 3.98MB/s] TOOL CODE: from smolagents import Tool from huggingface_hub import InferenceClient class TextToImageTool(Tool): description = "This tool creates an image according to a prompt, which is a text description." name = "image_generator" inputs = {"prompt": {"type": "string", "description": "The image generator prompt. Don't hesitate to add details in the prompt to make the image look better, like 'high-res, photorealistic', etc."}} output_type = "image" model_sdxl = "black-forest-labs/FLUX.1-schnell" client = InferenceClient(model_sdxl) def forward(self, prompt): return self.client.text_to_image(prompt) Traceback (most recent call last): File "/home/user/app/app.py", line 54, in <module> agent = CodeAgent( File "/usr/local/lib/python3.10/site-packages/smolagents/agents.py", line 805, in __init__ super().__init__( File "/usr/local/lib/python3.10/site-packages/smolagents/agents.py", line 148, in __init__ assert isinstance(tool, Tool), f"This element is not of class Tool: {str(tool)}" AssertionError: This element is not of class Tool: <module 'tools.visit_webpage' from '/home/user/app/tools/visit_webpage.py'>

Container logs: