storresbusquets commited on
Commit
b2ed0d8
·
1 Parent(s): a67ebcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ class GradioInference():
10
  self.current_size = "base"
11
  self.loaded_model = whisper.load_model(self.current_size)
12
  self.yt = None
13
- self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
14
 
15
  # Initialize VoiceLabT5 model and tokenizer
16
  self.keyword_model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
 
10
  self.current_size = "base"
11
  self.loaded_model = whisper.load_model(self.current_size)
12
  self.yt = None
13
+ self.summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
14
 
15
  # Initialize VoiceLabT5 model and tokenizer
16
  self.keyword_model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")