zeroshot
/

sst2-distilbert-sparse

Text Classification

Model card Files Files and versions Community

zeroshot commited on Nov 14, 2022

Commit

eaa4743

·

1 Parent(s): 8acb309

edit handler

Files changed (1) hide show

handler.py +1 -2

handler.py CHANGED Viewed

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

     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         """
         Args:
+            data (:obj:): prediction input text
         """
         inputs = data.pop("inputs", data)