Audio Classification
PyTorch
ONNX
whisper
biology
bnestor commited on
Commit
b644068
·
verified ·
1 Parent(s): 84ba99a

Upload handler.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -21,7 +21,7 @@ class EndpointHandler():
21
  model_id = 'DORI-SRKW/whisper-base-mm-cpu'
22
 
23
  # Load the model
24
- model = ORTModelForAudioClassification.from_pretrained(model_id, export=True)
25
  self.feature_extractor = WhisperFeatureExtractor.from_pretrained(model_id)
26
 
27
  self.model.eval()
 
21
  model_id = 'DORI-SRKW/whisper-base-mm-cpu'
22
 
23
  # Load the model
24
+ self.model = ORTModelForAudioClassification.from_pretrained(model_id, export=True)
25
  self.feature_extractor = WhisperFeatureExtractor.from_pretrained(model_id)
26
 
27
  self.model.eval()