Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import time
 from moviepy.editor import VideoFileClip
 from faster_whisper import WhisperModel
+from pytube import YouTube
 
 # 비디오를 MP3로 변환하는 함수
 def convert_mp4_to_mp3(video_file_path, output_dir):
@@ -38,10 +39,17 @@ def transcribe_audio(model_size, audio_file):
 
     return f"{detected_language}\n\nTranscription:\n{result_text}\n\nElapsed time: {elapsed_time:.2f} seconds"
 
+# YouTube URL에서 비디오를 다운로드하는 함수
+def download_youtube_video(url, output_dir):
+    yt = YouTube(url)
+    stream = yt.streams.filter(file_extension='mp4').first()
+    output_path = stream.download(output_dir)
+    return output_path
+
 # Gradio 인터페이스에서 사용할 메인 함수
 def process_video(model_size, video_file=None, video_url=None):
     if video_url and not video_file:
-        video_file_path = gr.processing_utils.download_url(video_url, dir='/tmp')
+        video_file_path = download_youtube_video(video_url, '/tmp')
     elif video_file and not video_url:
         video_file_path = video_file.name
     else: