update image code

app.py

@@ -16,18 +16,11 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def inference_image_tool(prompt:str, model:str = "black-forest-labs/FLUX.1-dev")-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """Generate an image from a text prompt using Hugging Face Inference Providers."
-    Args:
-        prompt: Text prompt for the image.
-        model: HF model id (e.g. black-forest-labs/FLUX.1-dev).
-    """
+def inference_image_tool(prompt: str, model: str = "black-forest-labs/FLUX.1-dev") -> "Image.Image":
     token = os.getenv("HF_TOKEN")
     if not token:
         raise RuntimeError("HF_TOKEN is missing. Add it in Space Settings → Secrets.")
-
-    client = InferenceClient('auto', api_key=token)
+    client = InferenceClient(provider="auto", api_key=token)
     return client.text_to_image(prompt, model=model)