Update pipeline.py

pipeline.py

@@ -297,19 +297,7 @@ class PreTrainedPipeline():
             model=self.model2, tokenizer=self.tokenizer2)
     
     def __call__(self, inputs: str)-> List[Dict[str, Any]]:
-
-        # TEMP testing
-        # data = [{"video_id": "pqh4LfPeCYs", "start": 835.933, "end": 927.581, "category": "sponsor"}]
-        # words = get_words("pqh4LfPeCYs")
-        # segment = extract_segment(words, 835.933, 927.581)
-        data = [{
-            'video_id': 'pqh4LfPeCYs',
-            'start': 835.933,
-            'end': 927.581,
-            'category': 'sponsor'
-        }]
-        # END TEMP
-
+        data = json.loads(inputs)
         return self.pipeline2(data)