Project Alpha 3

2024-10-10 11:03:15 -04:00
parent 1650f3fccb
commit d3e07ede69
2 changed files with 8 additions and 1 deletions
--- a/transcribe.py
+++ b/transcribe.py
@@ -15,6 +15,7 @@ def process(audio_path):
        tokenizer=processor.tokenizer,
        feature_extractor=processor.feature_extractor,
        torch_dtype=torch.float32,
+        return_timestamps=True,
        device="mps"
    )
    output = pipe(audio_path)["text"]