philschmid
/

t5-11b-sharded

Text2Text Generation

endpoints-template

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

philschmid HF staff commited on Oct 24, 2022

Commit

54d7f3a

•

1 Parent(s): 9f5bc78

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -30,4 +30,4 @@ class EndpointHandler:
         # postprocess the prediction
         prediction = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return {"generated_text": prediction}

         # postprocess the prediction
         prediction = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
+        return [{"generated_text": prediction}]