cguynup
/

flash_gen_bert_para

Inference Endpoints

Model card Files Files and versions Community

cguynup commited on Nov 22, 2023

Commit

59fca60

•

1 Parent(s): b6a854e

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -18,7 +18,7 @@ class EndpointHandler():
                 yield i
         labels = []
-        for out in pipeline(iterator(), padding=True, truncation=True, max_length=253):
             labels.append(int(out["label"][-1]))
         return {

                 yield i
         labels = []
+        for out in self.pipeline(iterator(), padding=True, truncation=True, max_length=253):
             labels.append(int(out["label"][-1]))
         return {