Update app.py

app.py

@@ -41,7 +41,7 @@ class GradioInference:
 
         # Perform summarization on the transcription
         transcription_summary = self.summarizer(
-            results["text"], max_length=130, min_length=30, do_sample=False
+            results["text"], max_length=512, min_length=30, do_sample=False
         )
 
         # Extract keywords using VoiceLabT5
@@ -84,7 +84,7 @@ class GradioInference:
 
         # Perform summarization on the transcription
         transcription_summary = self.summarizer(
-            results["text"], max_length=130, min_length=30, do_sample=False
+            results["text"], max_length=512, min_length=30, do_sample=False
         )
 
         # Extract keywords using VoiceLabT5
@@ -100,7 +100,7 @@ class GradioInference:
         keywords = [x.strip() for x in predicted.split(",") if x.strip()]
 
         label = self.classifier(results["text"])[0]["label"]
-        wordcloud = WordCloud(width=800, height=400, background_color="white").generate(
+        wordcloud = WordCloud().generate(
             results["text"]
         )
         wordcloud_image = wordcloud.to_image()
@@ -225,7 +225,7 @@ with block:
 
     with gr.Accordion("How does it work?", open=False):
         gr.Markdown(
-            "Works by using OpenAI's Whisper, DistilBART for summarization and VoiceLabT5 for Keyword Extraction."
+            "Works by using OpenAI's Whisper, BART for summarization and VoiceLabT5 for Keyword Extraction."
         )
 
     gr.HTML(