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, 4.42MB/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 42, in <module> GradioUI(agent).launch() NameError: name 'GradioUI' is not defined

Container logs: