Generate img return type

app.py

@@ -51,9 +51,9 @@ def generate_image_and_save(prompt: str) -> str:
     return f"Image sauvegardée dans : {filepath}"
 
 @tool
-def generate_image(prompt: str):
+def generate_image(prompt: str) -> object:
     """Génère une image à partir d'un prompt et la renvoie directement.
-    L'UI Gradio pourra l'afficher.
+    L'UI Gradio pourra l'afficher si elle gère les objets image.
     Args:
         prompt: description textuelle de l'image.
     """