Update app.py

app.py

@@ -10,7 +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="google/pegasus-large")
+        self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
                 
         # Initialize VoiceLabT5 model and tokenizer
         self.keyword_model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
@@ -126,7 +126,7 @@ with block as demo:
                 audio_file = gr.Audio(type="filepath")
                 with gr.Row().style(equal_height=True):
                     text = gr.Textbox(label="Transcription", placeholder="Transcription Output...", lines=10).style(show_copy_button=True, container=False)
-                # with gr.Row().style(equal_height=True):
+                with gr.Row().style(equal_height=True):
                     summary = gr.Textbox(label="Summary", placeholder="Summary Output", lines=5)
                     keywords = gr.Textbox(label="Keywords", placeholder="Keywords Output", lines=5)
                     label = gr.Label(label="Sentiment Analysis")