Update audio2text/init.py
Browse files- audio2text/init.py +1 -2
audio2text/init.py
CHANGED
|
@@ -7,11 +7,10 @@ from transformers import pipeline
|
|
| 7 |
|
| 8 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 9 |
|
| 10 |
-
checkpoint = "openai/whisper-
|
| 11 |
pipe = pipeline(
|
| 12 |
"automatic-speech-recognition",
|
| 13 |
model=checkpoint,
|
| 14 |
device=device,
|
| 15 |
chunk_length_s=30,
|
| 16 |
-
# stride_length_s=(4, 2),
|
| 17 |
)
|
|
|
|
| 7 |
|
| 8 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 9 |
|
| 10 |
+
checkpoint = "openai/whisper-medium"
|
| 11 |
pipe = pipeline(
|
| 12 |
"automatic-speech-recognition",
|
| 13 |
model=checkpoint,
|
| 14 |
device=device,
|
| 15 |
chunk_length_s=30,
|
|
|
|
| 16 |
)
|