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")
@@ -36,7 +36,7 @@ class GradioInference():
         
         results = self.loaded_model.transcribe(path, language=lang)
 
-        inputs = tokenizer(results["text"], max_length=1024, truncation=True, return_tensors="pt")
+        inputs = self.tokenizer_model(results["text"], max_length=1024, truncation=True, return_tensors="pt")
     
         summary_ids = self.keyword_model.generate(inputs["input_ids"])
         summary = self.keyword_tokenizer.batch_decode(summary_ids,