rheatony's picture
Update app.py
b3597f9
from youtube_transcript_api import YouTubeTranscriptApi
import gradio as gr
from gradio.mix import Series
def generate_transcript(url):
id = url[url.index("=")+1:]
transcript = YouTubeTranscriptApi.get_transcript(id)
script = ""
for text in transcript:
t = text["text"]
if t != '[Music]':
script += t + " "
return script
transcriber = gr.Interface(generate_transcript, 'text', 'text')
summarizer = gr.Interface.load("huggingface/sshleifer/distilbart-cnn-12-6")
gradio_ui = Series(transcriber, summarizer,
inputs = gr.inputs.Textbox(label = "Enter the YouTube URL:"),
outputs = gr.outputs.Textbox(label = "Transcript Summary"),
title = "Automatic Subtitles Generator",
theme = "peach",
description = "Automatic Subtitles Generator")
gradio_ui.launch()