CineAI commited on
Commit
dce4a52
1 Parent(s): 0ce306d

Update audio2text/init.py

Browse files
Files changed (1) hide show
  1. audio2text/init.py +1 -1
audio2text/init.py CHANGED
@@ -9,7 +9,7 @@ device = 0 if torch.cuda.is_available() else "cpu"
9
 
10
  checkpoint = "openai/whisper-base"
11
  pipe = pipeline(
12
- task="automatic-speech-recognition",
13
  model=checkpoint,
14
  device=device,
15
  )
 
9
 
10
  checkpoint = "openai/whisper-base"
11
  pipe = pipeline(
12
+ "automatic-speech-recognition",
13
  model=checkpoint,
14
  device=device,
15
  )