Update app.py

app.py

@@ -3,32 +3,31 @@ from youtube_transcript_api import YouTubeTranscriptApi
 import gradio as gr
 
 def summarize(youtube_video):
-
   video_id = youtube_video.split("=")[1]
   transcript = YouTubeTranscriptApi.get_transcript(video_id)
-  
-  summarizer = pipeline("summarization", model="knkarthick/MEETING_SUMMARY")
-  result = ""
+
+  summarizer = pipeline('summarization',model='sshleifer/distilbart-cnn-12-6')
+
+  input_text = ""
   for i in transcript:
-      result += ' ' + i['text']
+      input_text += ' ' + i['text']
 
-  num_iters = int(len(result)/2000)
+  num_iters = int(len(input_text)/1000)
   summarized_text = []
   for i in range(0, num_iters + 1):
     start = 0
-    start = i * 2000
-    end = (i + 1) * 2000
-    print("input text \n" + result[start:end])
-    out = summarizer(result[start:end])
+    start = i * 1000
+    end = (i + 1) * 1000
+    print("input text \n" + input_text[start:end])
+    out = summarizer(input_text[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
+    # inp = input_text.replace('\n','')
+  output_text=' '.join(summarized_text)
+  return output_text
 
 App= gr.Interface(fn=summarize, inputs="text", outputs="text")
 App.launch()