Update app.py

app.py

@@ -3,9 +3,14 @@ from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
 from transformers import pipeline
 import re
-import gradio
+import gradio as gr
 
-pipe = pipeline("summarization", model="Falconsai/text_summarization")
+pipe = pipeline("summarization", model="Falconsai/text_summarization", device=-1)
+
+def chunk_length(text, max_length = 1024):
+    words = text.split()
+    for i in range(0, len(words), max_length)
+    yield " ".join(words[i:i + max_length])
 
 def extract_youtube_id(url):
     """
@@ -26,8 +31,12 @@ def extract_youtube_id(url):
     return None
 
 def summary(text_transcript):
-    output = pipe(text_transcript)
-    return output[0]['summary_text']
+    try:
+        chunks = chunk_text(text_transcript)
+        summarized_chunks = [pipe(chunk)[0]['summary_text'] for chunk in chunks]
+        return " ".join(summarized_chunks)
+    except Exception as e:
+        return f"Error during summarization: {e}"
 
 def get_youtube_transcript(link):
     video_id= extract_youtube_id(link)