Porjaz commited on
Commit
440f4a4
·
verified ·
1 Parent(s): 4902d74

Update custom_interface_app.py

Browse files
Files changed (1) hide show
  1. custom_interface_app.py +3 -3
custom_interface_app.py CHANGED
@@ -161,8 +161,8 @@ class ASR(Pretrained):
161
 
162
  # Pass the segment through the ASR model
163
  result = " ".join(self.encode_batch_w2v2(device, batch, rel_length)[0])
164
- outputs.append(result)
165
- return outputs
166
  else:
167
  waveform = torch.tensor(waveform).to(device)
168
  waveform = waveform.to(device)
@@ -170,7 +170,7 @@ class ASR(Pretrained):
170
  batch = waveform.unsqueeze(0)
171
  rel_length = torch.tensor([1.0]).to(device)
172
  outputs = " ".join(self.encode_batch_w2v2(device, batch, rel_length)[0])
173
- return [outputs]
174
 
175
 
176
 
 
161
 
162
  # Pass the segment through the ASR model
163
  result = " ".join(self.encode_batch_w2v2(device, batch, rel_length)[0])
164
+ # outputs.append(result)
165
+ yield result
166
  else:
167
  waveform = torch.tensor(waveform).to(device)
168
  waveform = waveform.to(device)
 
170
  batch = waveform.unsqueeze(0)
171
  rel_length = torch.tensor([1.0]).to(device)
172
  outputs = " ".join(self.encode_batch_w2v2(device, batch, rel_length)[0])
173
+ yield [outputs]
174
 
175
 
176