Update app.py

app.py

@@ -25,7 +25,7 @@ yt_video_dir = "./yt_dir/0"
 os.makedirs(yt_video_dir, exist_ok=True)
 
 
-def get_audio_from_yt_video_input(yt_link: str):
+def get_audio_from_yt_video_input(yt_link: str, start_point_in_second=0, duration_in_second=30):
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
@@ -35,9 +35,12 @@ def get_audio_from_yt_video_input(yt_link: str):
         warnings.warn(f"Video Not Found at {yt_link} ({e})")
         filename_in = None
 
+    # trim audio length - due to computation time on HuggingFace environment
+    trim_audio(target_file_path=filename_in, start_point_in_second=start_point_in_second, duration_in_second=duration_in_second)
+    
     return filename_in, filename_in
 
-def get_audio_from_yt_video_ref(yt_link: str):
+def get_audio_from_yt_video_ref(yt_link: str, start_point_in_second=0, duration_in_second=30):
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
@@ -47,6 +50,9 @@ def get_audio_from_yt_video_ref(yt_link: str):
         warnings.warn(f"Video Not Found at {yt_link} ({e})")
         filename_ref = None
 
+    # trim audio length - due to computation time on HuggingFace environment
+    trim_audio(target_file_path=filename_ref, start_point_in_second=start_point_in_second, duration_in_second=duration_in_second)
+    
     return filename_ref, filename_ref
 
 def inference(file_uploaded_in, file_uploaded_ref):