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="sshleifer/distilbart-cnn-12-6")
+        # self.summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
         self.tokenizer_model = AutoTokenizer.from_pretrained("google/pegasus-large")
         self.summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-large")
@@ -44,7 +44,7 @@ class GradioInference():
                                   clean_up_tokenization_spaces=False)
         
         # Perform summarization on the transcription
-        transcription_summary = self.summarizer(results["text"], max_length=130, min_length=30, do_sample=False)
+        # transcription_summary = self.summarizer(results["text"], max_length=130, min_length=30, do_sample=False)
 
         # Extract keywords using VoiceLabT5
         task_prefix = "Keywords: "