psakamoori
/

speechbrain-emotion-recognition-openvino

Audio Classification

Model card Files Files and versions

psakamoori commited on Jun 11, 2024

Commit

36a39af

·

1 Parent(s): b02e0f2

Update custom_interface.py

Files changed (1) hide show

custom_interface.py +0 -2

custom_interface.py CHANGED Viewed

@@ -202,8 +202,6 @@ class CustomEncoderWav2vec2Classifier(Pretrained):
     def ov_inference(self, wavs, wav_lens):
         output_tensor = self.compiled_model(wavs.float())[0]
         output_tensor = torch.from_numpy(output_tensor)
-        print("\n[INFO] Performing OpenVINO inference...")
         return output_tensor
     def forward(self, wavs, wav_lens=None, normalize=False):

     def ov_inference(self, wavs, wav_lens):
         output_tensor = self.compiled_model(wavs.float())[0]
         output_tensor = torch.from_numpy(output_tensor)
         return output_tensor
     def forward(self, wavs, wav_lens=None, normalize=False):