Spaces:
Runtime error
Runtime error
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() |