Xenova
/

sponsorblock-classifier-v2

Text Classification

Model card Files Files and versions Community

Joshua Lochner commited on Apr 13, 2022

Commit

34790a9

•

1 Parent(s): 8cb1232

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -297,8 +297,8 @@ class SponsorBlockClassificationPipeline(TextClassificationPipeline):
         results = super().postprocess(model_outputs, function_to_apply, return_all_scores)
-        # for result in results:
-        #     result['label_text'] = CATEGORIES[result['label']]
         return results # {**data['video'], 'result': results}

         results = super().postprocess(model_outputs, function_to_apply, return_all_scores)
+        for result in results:
+            result['label_text'] = CATEGORIES[result['label']]
         return results # {**data['video'], 'result': results}