Reverting to the old thingy

grand-cat-unknown · Mar 29, 2024 · ebe7d44 · ebe7d44
1 parent e49039f
commit ebe7d44
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/builder/requirements.txt b/builder/requirements.txt
@@ -7,5 +7,4 @@
 runpod==1.6.2
 python-dotenv==0.19.0
 whisper-timestamped
-stable-ts
-demucs
+stable-ts
diff --git a/src/handler.py b/src/handler.py
@@ -20,15 +20,15 @@ def handler(job):
 
     print("Downloading song from s3")
     s3.download_file('auto-karaoke', f'{song_name}/vocals.mp3', 'vocals.mp3')
-    s3.download_file('auto-karaoke', f'{song_name}/base_song.mp3', 'base_song.mp3')
+    # s3.download_file('auto-karaoke', f'{song_name}/base_song.mp3', 'base_song.mp3')
 
     s3.download_file('auto-karaoke', f'{song_name}/lyrics.txt', 'lyrics.txt')
 
     with open('lyrics.txt', 'r') as f:
         lyrics = f.read()
 
-    # result = model.align('vocals.mp3', lyrics, 'English')
-    result = model.align('base_song.mp3', lyrics, 'English',denoiser="demucs", vad=True)
+    result = model.align('vocals.mp3', lyrics, 'English')
+    # result = model.align('base_song.mp3', lyrics, 'English',denoiser="demucs", vad=True)
     model.refine('audio.mp3', result)
     # audio = stable_whisper.load_audio("vocals.mp3")
     # print("transcribing audio")