|
from youtube_transcript_api import YouTubeTranscriptApi |
|
from youtube_transcript_api.formatters import TextFormatter |
|
import gradio as gr |
|
from gradio.mix import Series |
|
|
|
def generate_transcript(url): |
|
|
|
text = url[url.index("=")+1:] |
|
transcript = YouTubeTranscriptApi.get_transcript(text) |
|
|
|
formatter = TextFormatter() |
|
text_formatted = formatter.format_transcript(transcript) |
|
|
|
return text_formatted |
|
|
|
transcriber = gr.Interface(generate_transcript, 'text', 'text') |
|
summarizer = gr.Interface.load("huggingface/facebook/bart-large-cnn") |
|
|
|
description = ''' |
|
This application summarizes a YouTube video based on its transcript. |
|
Enter a Youtube video URL and click submit for obtaining a summary. |
|
''' |
|
|
|
youtube_url_examples = [['https://www.youtube.com/watch?v=MBRqu0YOH14'], |
|
['https://www.youtube.com/watch?v=UjtOGPJ0URM']] |
|
|
|
|
|
|
|
iface = Series(transcriber, summarizer, |
|
inputs = gr.inputs.Textbox(label = "Enter the YouTube URL"), |
|
outputs = gr.outputs.Textbox(label = "Transcript Summary"), |
|
examples = youtube_url_examples, |
|
title = "Distil YouTube", |
|
theme = "grass", |
|
description = description) |
|
|
|
iface.launch() |
