Spaces:
Runtime error
Runtime error
| # pip install -q transformers | |
| # pip install -q youtube_transcript_api | |
| # pip install -q gradio | |
| import gradio as gr | |
| from transformers import pipeline | |
| from youtube_transcript_api import YouTubeTranscriptApi | |
| def summarize_transcript(youtube_video): | |
| video_id = youtube_video.split("=")[1] | |
| transcript = YouTubeTranscriptApi.get_transcript(video_id) | |
| result = "" | |
| for i in transcript: | |
| result += ' ' + i['text'] | |
| summarizer = pipeline('summarization') | |
| num_iters = int(len(result) / 1000) | |
| summarized_text = "" | |
| for i in range(0, num_iters + 1): | |
| start = i * 1000 | |
| end = (i + 1) * 1000 | |
| out = summarizer(result[start:end]) | |
| out = out[0] | |
| out = out['summary_text'] | |
| summarized_text += out | |
| return summarized_text | |
| iface = gr.Interface(fn=summarize_transcript, inputs="text", outputs="text") | |
| iface.launch() | |