Update models.py
Browse files
models.py
CHANGED
|
@@ -121,7 +121,7 @@ class ModelManager:
|
|
| 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)
|
|
|
|
| 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)
|