From 162711ebeb4a69dbe92f5218df8c4e56ab045b37 Mon Sep 17 00:00:00 2001 From: Izaya Orihara Date: Fri, 6 Oct 2023 16:53:57 +0200 Subject: [PATCH] Check-out: test of code --- README.md | 1 + requirements.txt | 2 +- src/main.py | 3 +-- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0fb7ac8..a2f259b 100644 --- a/README.md +++ b/README.md @@ -7,6 +7,7 @@ Transcriber code ## to-do - seperate data production form code +- seperate huggingface (hf) token out ``` project_root/ diff --git a/requirements.txt b/requirements.txt index 2300225..8ed7f28 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ youtube-dl -pfy pytube pydub +torch git+https://github.com/m-bain/whisperx.git \ No newline at end of file diff --git a/src/main.py b/src/main.py index 9a3a7b0..50e0033 100644 --- a/src/main.py +++ b/src/main.py @@ -4,8 +4,7 @@ from src.transcript_generator import TranscriptGenerator def main(): # input could be a YouTube link or a local file input_sources = [ - 'https://www.youtube.com/watch?v=BVVo1BaL6JA', - '/path/to/your/local/file.mp3' + 'https://www.youtube.com/watch?v=BVVo1BaL6JA' ] transcript_generator = TranscriptGenerator()