psakamoori
commited on
Commit
•
36a39af
1
Parent(s):
b02e0f2
Update custom_interface.py
Browse files- custom_interface.py +0 -2
custom_interface.py
CHANGED
@@ -202,8 +202,6 @@ class CustomEncoderWav2vec2Classifier(Pretrained):
|
|
202 |
def ov_inference(self, wavs, wav_lens):
|
203 |
output_tensor = self.compiled_model(wavs.float())[0]
|
204 |
output_tensor = torch.from_numpy(output_tensor)
|
205 |
-
print("\n[INFO] Performing OpenVINO inference...")
|
206 |
-
|
207 |
return output_tensor
|
208 |
|
209 |
def forward(self, wavs, wav_lens=None, normalize=False):
|
|
|
202 |
def ov_inference(self, wavs, wav_lens):
|
203 |
output_tensor = self.compiled_model(wavs.float())[0]
|
204 |
output_tensor = torch.from_numpy(output_tensor)
|
|
|
|
|
205 |
return output_tensor
|
206 |
|
207 |
def forward(self, wavs, wav_lens=None, normalize=False):
|