remove returns_timestamp

Files changed (2) hide show

__pycache__/handler.cpython-39.pyc CHANGED Viewed

Binary files a/__pycache__/handler.cpython-39.pyc and b/__pycache__/handler.cpython-39.pyc differ

handler.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from typing import Dict, List, Any
-from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
 import torch
 class EndpointHandler():
@@ -22,5 +22,5 @@ class EndpointHandler():
         """
         inputs = data.pop("inputs",data)
         print("inputs", inputs)
-        prediction = self.pipe(inputs, return_timestamps=True)
         return prediction

 from typing import Dict, List, Any
+from transformers import pipeline
 import torch
 class EndpointHandler():
         """
         inputs = data.pop("inputs",data)
         print("inputs", inputs)
+        prediction = self.pipe(inputs)
         return prediction