File size: 882 Bytes
475d3ba
 
 
3192dea
 
 
 
4a055e9
 
3192dea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0873811
 
 
 
 
d69eda7
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
34
35
from transformers import pipeline
from youtube_transcript_api import YouTubeTranscriptApi

def summarize(youtube_video):

  video_id = youtube_video.split("=")[1]
  transcript = YouTubeTranscriptApi.get_transcript(video_id)
  
  summarizer = pipeline('summarization')

  result = ""
  for i in transcript:
      result += ' ' + i['text']

  num_iters = int(len(result)/1000)
  summarized_text = []
  for i in range(0, num_iters + 1):
    start = 0
    start = i * 1000
    end = (i + 1) * 1000
    print("input text \n" + result[start:end])
    out = summarizer(result[start:end])
    out = out[0]
    out = out['summary_text']
    print("Summarized text\n"+out)
    summarized_text.append(out)

    inp = result.replace('\n','')
    summary = summarizer(inp)
    return summary
    
import gradio as gr

grad= gr.Interface(fn=summarize, inputs="text", outputs="text")
grad.launch()