Update app.py

app.py

@@ -3,6 +3,7 @@ from pytube import YouTube
 from transformers import pipeline
 import gradio as gr
 import os
+import re
 
 model = whisper.load_model("base")
 summarizer = pipeline("summarization")
@@ -22,9 +23,8 @@ def get_text(url):
   return result['text']
 
 def get_summary(article):
-  print(article)
-  b = summarizer(article, min_length=20, max_length=30, do_sample=False)
-  print(b)
+  first_sentences = ' '.join(re.split(r'(?<=[.:;])\s', article)[:5])
+  b = summarizer(first_sentences, min_length=20, max_length=30, do_sample=False)
   b = b[0]['summary_text']
   return b