philschmid
/

flan-t5-xxl-sharded-fp16

Text2Text Generation Transformers PyTorch t5 endpoints-template Inference Endpoints text-generation-inference

Model card Files Files and versions Community

Update handler.py

#4

by lewtun HF staff - opened Mar 8, 2023

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -11,8 +11,8 @@ class EndpointHandler:
     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         """
         Args:
-            data (:obj:):
-                includes the deserialized image file as PIL.Image
         """
         # process input
         inputs = data.pop("inputs", data)

     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         """
         Args:
+            data (:dict:):
+                The payload with the text prompt and generation parameters.
         """
         # process input
         inputs = data.pop("inputs", data)