Update app.py

app.py

@@ -21,7 +21,7 @@ from pytube.exceptions import VideoUnavailable
 from inference.style_transfer import *
 
 
-yt_video_dir = "./yt_dir"
+yt_video_dir = "./yt_dir/0"
 os.makedirs(yt_video_dir, exist_ok=True)
 
 
@@ -29,7 +29,7 @@ def get_audio_from_yt_video_input(yt_link: str):
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
-        filename = os.path.join(yt_video_dir, "input.wav")
+        filename = os.path.join(yt_video_dir, "input.mp3")
         t[0].download(filename=filename)
     except VideoUnavailable as e:
         warnings.warn(f"Video Not Found at {yt_link} ({e})")
@@ -41,7 +41,7 @@ def get_audio_from_yt_video_ref(yt_link: str):
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
-        filename = os.path.join(yt_video_dir, "reference.wav")
+        filename = os.path.join(yt_video_dir, "reference.mp3")
         t[0].download(filename=filename)
     except VideoUnavailable as e:
         warnings.warn(f"Video Not Found at {yt_link} ({e})")