Update app.py

app.py

@@ -1,9 +1,32 @@
 !pip install -q transformers
-!pip install -q youtube_transcript_api 
+!pip install -q youtube_transcript_api
+!pip install -q gradio
+
+
+import gradio as gr
 from transformers import pipeline
-from youtube_transcript_api import YouTubeTranscriptApi 
+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()
 
-youtube_video = 'https://www.youtube.com/watch?v=lv1_-RER4_I'
-video_id = youtube_video.split("=")[1]
-from IPython.display import YouTubeVideo
-YouTubeVideo(video_id)