Xenova
/

sponsorblock-classifier-v2

Text Classification

Model card Files Files and versions Community

Joshua Lochner commited on Apr 13, 2022

Commit

1581175

•

1 Parent(s): dd33b0c

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -314,8 +314,8 @@ class SponsorBlockClassificationPipeline(TextClassificationPipeline):
         if isinstance(data, str): # If string, assume this is what user wants to classify
             text = data
         else: # Otherwise, get data from transcript
-            words = get_words(video['video_id'])
-            segment_words = extract_segment(words, video['start'], video['end'])
             text = ' '.join(x['text'] for x in segment_words)
         return self.tokenizer(

         if isinstance(data, str): # If string, assume this is what user wants to classify
             text = data
         else: # Otherwise, get data from transcript
+            words = get_words(data['video_id'])
+            segment_words = extract_segment(words, data['start'], data['end'])
             text = ' '.join(x['text'] for x in segment_words)
         return self.tokenizer(