Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import whisper
 from pytube import YouTube
-
+from transformers import pipeline
 
 class GradioInference():
     def __init__(self):
@@ -10,6 +10,7 @@ class GradioInference():
         self.current_size = "base"
         self.loaded_model = whisper.load_model(self.current_size)
         self.yt = None
+        self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
     def __call__(self, link, lang, size):
         if self.yt is None:
@@ -24,7 +25,10 @@ class GradioInference():
             self.current_size = size
         results = self.loaded_model.transcribe(path, language=lang)
 
-        return results["text"]
+        # Perform summarization on the transcription
+        transcription_summary = self.summarizer(results["text"], max_length=130, min_length=30, do_sample=False)
+
+        return results["text"], transcription_summary[0]["summary_text"]
 
     def populate_metadata(self, link):
         self.yt = YouTube(link)
@@ -33,7 +37,7 @@ class GradioInference():
 
 gio = GradioInference()
 title = "Youtube Insights"
-description = "The All-in-One Video Analytics Tool"
+description = "Your AI-powered video analytics tool"
 
 block = gr.Blocks()
 with block:
@@ -44,7 +48,7 @@ with block:
             <h1>Youtube Insights</h1>
           </div>
           <p style="margin-bottom: 10px; font-size: 94%">
-            The All-in-One Video Analytics Tool
+            Your AI-powered video analytics tool
           </p>
         </div>
         """
@@ -59,9 +63,11 @@ with block:
             with gr.Row().style(equal_height=True):
                 img = gr.Image(label="Thumbnail")
                 text = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
+                summary = gr.Textbox(label="Summary", placeholder="Summary Output", lines=5)  # Added summary output
             with gr.Row().style(equal_height=True):
-                btn = gr.Button("Transcribe")
-            btn.click(gio, inputs=[link, lang, sz], outputs=[text])
+                btn = gr.Button("Get video insights")  # Updated button label
+            btn.click(gio, inputs=[link, lang, sz], outputs=[text, summary])  # Added summary output
             link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 
 block.launch()
+