Update tools/image_caption.py

tools/image_caption.py

@@ -1,6 +1,8 @@
 from smolagents import Tool
 from huggingface_hub import InferenceClient
+from PIL import Image
 import requests
+from io import BytesIO
 
 class ImageCaptioningTool(Tool):
     description = "Cet outil génère une légende descriptive pour une image donnée."
@@ -11,5 +13,9 @@ class ImageCaptioningTool(Tool):
     client = InferenceClient(model_id)
 
     def forward(self, image_url):
-        image = requests.get(image_url, stream=True).raw
-        return self.client.image_to_text(image)
+        response = requests.get(image_url)
+        if response.status_code == 200:
+            image = Image.open(BytesIO(response.content))
+            return self.client.image_to_text(image)
+        else:
+            return f"Erreur lors du téléchargement de l'image : {response.status_code}"