Sjanmanchi commited on
Commit
178336a
·
verified ·
1 Parent(s): 7e61fd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import datetime
10
  # Load Hugging Face models
11
  device = 0 if torch.cuda.is_available() else -1
12
  whisper = pipeline("automatic-speech-recognition", model="openai/whisper-large", device=device)
13
- punctuate = pipeline("text2text-generation", model="bqho/bert-punctuator")
14
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
15
 
16
  # Utility: Extract audio
 
10
  # Load Hugging Face models
11
  device = 0 if torch.cuda.is_available() else -1
12
  whisper = pipeline("automatic-speech-recognition", model="openai/whisper-large", device=device)
13
+ punctuate = pipeline("text2text-generation", model="flexudy/t5-small-wav2vec2-grammar-fixer")
14
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
15
 
16
  # Utility: Extract audio