Xenova
/

sponsorblock-classifier-v2

Text Classification

Model card Files Files and versions Community

Joshua Lochner commited on Apr 13, 2022

Commit

fcc71d5

•

1 Parent(s): 9c77c04

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -310,7 +310,7 @@ class PreTrainedPipeline():
             'end': 927.581,
             'category': 'sponsor'
         }]
-        segment = self.pipeline2(data)
         # END TEMP
@@ -367,6 +367,7 @@ class PreTrainedPipeline():
             # widget needs the below format, for each class we return label and mask string
             labels.append({
                 "label": f"LABEL_{cls}",
                 "mask": mask_codes[f"mask_{cls}"],
                 "score": 1.0,

             'end': 927.581,
             'category': 'sponsor'
         }]
+        segment = data #self.pipeline2(data)
         # END TEMP
             # widget needs the below format, for each class we return label and mask string
             labels.append({
+                'inputs': inputs,
                 "label": f"LABEL_{cls}",
                 "mask": mask_codes[f"mask_{cls}"],
                 "score": 1.0,