Upload tool

requirements.txt

@@ -1,3 +1,3 @@
-Pillow
+PIL
 requests
 smolagents

tool.py

@@ -7,7 +7,7 @@ class SimpleTool(Tool):
     inputs = {'prompt': {'type': 'string', 'description': 'The prompt to generate the image from'}, 'width': {'type': 'integer', 'nullable': True, 'description': 'image width'}, 'height': {'type': 'integer', 'nullable': True, 'description': 'image height'}, 'seed': {'type': 'integer', 'nullable': True, 'description': 'A decimal integer  Each seed generates a new image variation Returns generated image as a PILL image'}}
     output_type = "object"
 
-    def forward(self, prompt: str, width: int = 1024, height: int = 1024, seed: int = 0) -> PIL.Image.Image:
+    def forward(self, prompt: str, width: int = 1024, height: int = 1024, seed: int = 0) -> 'PIL.Image.Image':
       """
       Generate an image from a prompt
          Args:
@@ -28,7 +28,7 @@ class SimpleTool(Tool):
         seed=random.randint(1,8*1024**2)
       else:
         random.seed(seed)
-      prompt="A beautiful landscape"
+
       url = f"https://pollinations.ai/p/{prompt}"
       params={"width":width,"height":height,"seed":seed}
       response = requests.get(url)