File size: 882 Bytes
02d1e8d
 
 
9377d92
 
 
fb627cb
9377d92
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fb627cb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# 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()