Aktraiser
/

modele-test

Text Generation

génération-de-texte

8-bit precision

Model card Files Files and versions

Aktraiser commited on Nov 27, 2024

Commit

37b22c3

·

verified ·

1 Parent(s): 9823ac3

Update handler.py

Files changed (1) hide show

handler.py +4 -4

handler.py CHANGED Viewed

@@ -48,10 +48,10 @@ class EndpointHandler:
                 **generation_kwargs
             )
-            # Formater la sortie
             if isinstance(outputs, list):
-                return {"generated_text": outputs[0]["generated_text"]}
-            return {"generated_text": outputs["generated_text"]}
         except Exception as e:
-            return {"error": str(e)}

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