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

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="flexudy/t5-small-wav2vec2-grammar-fixer")
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="vennify/t5-base-grammar-correction")
14
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
15
 
16
  # Utility: Extract audio