Update handler.py

handler.py

@@ -48,10 +48,10 @@ class EndpointHandler:
                 **generation_kwargs
             )
 
-            # Formater la sortie
+            # Formater la sortie en tableau comme requis par l'API
             if isinstance(outputs, list):
-                return {"generated_text": outputs[0]["generated_text"]}
-            return {"generated_text": outputs["generated_text"]}
+                return [{"generated_text": output["generated_text"]} for output in outputs]
+            return [{"generated_text": outputs["generated_text"]}]
 
         except Exception as e:
-            return {"error": str(e)}
+            return [{"error": str(e)}]