Bazedgul's picture
Update app.py
02d1e8d
raw
history blame contribute delete
882 Bytes
# 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()