Update handler.py

handler.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 import holidays
 
 class EndpointHandler():
-    def __init__(self, path="Evan-Lin/whisper-large-tw"):
-        self.pipeline = pipeline(model=path)
+    def __init__(self, path="."):
+        self.pipeline = pipeline(task="automatic-speech-recognition", model=path)
         self.pipeline.model.config.forced_decoder_ids = self.pipeline.model.processor.get_decoder_prompt_ids(language="Chinese", task="transcribe")
     	self.pipeline.model.generation_config.forced_decoder_ids = self.pipeline.model.config.forced_decoder_ids  # just to be sure!
 
@@ -20,5 +20,5 @@ class EndpointHandler():
 
         # run normal prediction
         inputs = data.pop("inputs",data)
-        prediction = self.pipe(inputs, return_timestamps=False)
+        prediction = self.pipeline(inputs, return_timestamps=False)
         return prediction