lpw commited on
Commit
1e62d89
1 Parent(s): 2225f99

Update audio_pipe.py

Browse files
Files changed (1) hide show
  1. audio_pipe.py +1 -0
audio_pipe.py CHANGED
@@ -124,6 +124,7 @@ class SpeechToSpeechPipeline():
124
  # _inputs = torch.from_numpy(inputs).unsqueeze(0)
125
  # print(f"input: {inputs}")
126
  # _inputs = torchaudio.load(inputs)
 
127
  sample, text = None, None
128
  if self.cfg.task._name in ["speech_to_text", "speech_to_text_sharded"]:
129
  sample = S2THubInterface.get_model_input(self.task, _inputs)
 
124
  # _inputs = torch.from_numpy(inputs).unsqueeze(0)
125
  # print(f"input: {inputs}")
126
  # _inputs = torchaudio.load(inputs)
127
+ _inputs = inputs
128
  sample, text = None, None
129
  if self.cfg.task._name in ["speech_to_text", "speech_to_text_sharded"]:
130
  sample = S2THubInterface.get_model_input(self.task, _inputs)