Update handler.py

handler.py

@@ -15,17 +15,17 @@ class EndpointHandler:
         self.tokenizer = WhisperTokenizer.from_pretrained('openai/whisper-large', language="korean", task='transcribe')        
         self.model = WhisperForConditionalGeneration.from_pretrained(path)
         #self.tokenizer = WhisperTokenizer.from_pretrained(path)
-        self.processor = WhisperProcessor.from_pretrained(path, language="korean", task='transcribe')
+        #self.processor = WhisperProcessor.from_pretrained(path, language="korean", task='transcribe')
         #self.processor = AutoProcessor.from_pretrained(path)
         #self.pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=processor.feature_extractor, feature_extractor=processor.feature_extractor)
-        self.feature_extractor = WhisperFeatureExtractor.from_pretrained(path)
+        self.feature_extractor = WhisperFeatureExtractor.from_pretrained('openai/whisper-large')
 
         
         # Move model to device
 #        self.model.to(device)
         
     def __call__(self, data: Any) -> List[Dict[str, str]]:
-        print('HELLO')
+        print('==========NEW PROCESS=========')
         #print(f"{data}")
         #inputs = data.pop("inputs", data)
         #print(f'1. inputs: {inputs}')
@@ -33,13 +33,13 @@ class EndpointHandler:
         
         inputs, _ = sf.read(io.BytesIO(data['inputs']))
         #inputs, _ = sf.read(data['inputs'])
-        print(f'2. inputs: {inputs}')
+        #print(f'2. inputs: {inputs}')
         
         input_features = self.feature_extractor(inputs, sampling_rate=16000).input_features[0]
-        print(f'3. input_features: {input_features}')
+        #print(f'3. input_features: {input_features}')
         input_features_tensor = torch.tensor(input_features).unsqueeze(0)
         input_ids = self.model.generate(input_features_tensor)
-        print(f'4. input_ids: {input_ids}')
+        #(f'4. input_ids: {input_ids}')
 
         transcription = self.tokenizer.batch_decode(input_ids, skip_special_tokens=True)[0]