Spaces:
Running
Running
| import gradio as gr | |
| import whisper | |
| import os | |
| model = whisper.load_model("tiny") | |
| # main processing | |
| def main(URL): | |
| video_path = youtube_dl(URL) | |
| text = transcript(video_path) | |
| return text | |
| # Transcribe text using Whisper | |
| def transcript(video_path): | |
| result = model.transcribe(video_path) | |
| return result["text"] | |
| # Download the video | |
| def youtube_dl(URL): | |
| # Download the video | |
| os.system(f"yt-dlp -f best -v {URL} -o target.mp4") | |
| # Path of downloaded video | |
| video_path = os.path.join(os.path.dirname(__file__), "target.mp4") | |
| return video_path | |
| with gr.Blocks() as demo: | |
| gr.Markdown( | |
| """ | |
| # YouTube video transcription app | |
| Enter the YouTube URL, press the button, and it will transcribe the video. | |
| """ | |
| ) | |
| with gr.Row(): | |
| with gr.Column(): | |
| with gr.Row(): | |
| url = gr.Textbox(placeholder = 'Youtube video URL', label = 'URL') | |
| with gr.Row(): | |
| transcribe_btn = gr.Button('Transcribe') | |
| with gr.Column(): | |
| outputs = gr.Textbox(placeholder = 'Transcription of the video', label = 'Transcription') | |
| transcribe_btn.click(fn = main, | |
| inputs = url, | |
| outputs = outputs | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |