EYEDOL commited on
Commit
db24fa9
·
verified ·
1 Parent(s): 0115b20

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -121,7 +121,7 @@ class ModelManager:
121
  )
122
 
123
  with torch.no_grad():
124
- ids = self.stt_model.generate(features, language="en", task="transcribe")
125
 
126
  text = self.stt_processor.batch_decode(ids, skip_special_tokens=True)[0].strip()
127
  logger.info("STT (%.0f ms): '%s'", (time.perf_counter() - t0) * 1000, text)
 
121
  )
122
 
123
  with torch.no_grad():
124
+ ids = self.stt_model.generate(features)
125
 
126
  text = self.stt_processor.batch_decode(ids, skip_special_tokens=True)[0].strip()
127
  logger.info("STT (%.0f ms): '%s'", (time.perf_counter() - t0) * 1000, text)