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="philschmid/bart-large-cnn-samsum")
+        self.summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
         
         # Initialize VoiceLabT5 model and tokenizer
         self.keyword_model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
@@ -92,7 +92,7 @@ with block as demo:
                     keywords = gr.Textbox(label="Keywords", placeholder="Keywords Output", lines=5).style(show_copy_button=True, container=True)
                     label = gr.Label(label="Sentiment Analysis")
                 with gr.Row().style(equal_height=True):
-                    clear = gr.ClearButton([size, lang, link, title, img, text, summary, keywords, label])
+                    clear = gr.ClearButton([link, title, img, text, summary, keywords, label])
                     btn = gr.Button("Get video insights")  # Updated button label
                 btn.click(gio, inputs=[link, lang, size], outputs=[text, summary, keywords, label])
                 link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
@@ -110,7 +110,7 @@ with block as demo:
                 #     summary = gr.Textbox(label="Summary", placeholder="Summary Output", lines=5)
                 #     keywords = gr.Textbox(label="Keywords", placeholder="Keywords Output", lines=5)
                 with gr.Row().style(equal_height=True):
-                    clear = gr.ClearButton([size, lang, text])
+                    clear = gr.ClearButton([text])
                     btn = gr.Button("Get video insights")  # Updated button label
                 btn.click(transcribe_audio, inputs=[audio_file], outputs=[text])
                 # link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])