wtlow003
commited on
Commit
·
48dd2f7
1
Parent(s):
365de93
fix: model
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ from transformers import pipeline
|
|
| 7 |
from transformers.pipelines.audio_utils import ffmpeg_read
|
| 8 |
|
| 9 |
MODEL_NAME = "jensenlwt/whisper-small-singlish-122k"
|
| 10 |
-
BATCH_SIZE = 8
|
| 11 |
FILE_LIMIT_MB = 1000
|
| 12 |
|
| 13 |
device = 0 if torch.cuda.is_available() else "cpu"
|
|
@@ -28,7 +27,6 @@ def transcribe(inputs, task):
|
|
| 28 |
|
| 29 |
text = pipe(
|
| 30 |
inputs,
|
| 31 |
-
batch_size=BATCH_SIZE,
|
| 32 |
generate_kwargs={"task": task},
|
| 33 |
return_timestamps=True,
|
| 34 |
)["text"]
|
|
|
|
| 7 |
from transformers.pipelines.audio_utils import ffmpeg_read
|
| 8 |
|
| 9 |
MODEL_NAME = "jensenlwt/whisper-small-singlish-122k"
|
|
|
|
| 10 |
FILE_LIMIT_MB = 1000
|
| 11 |
|
| 12 |
device = 0 if torch.cuda.is_available() else "cpu"
|
|
|
|
| 27 |
|
| 28 |
text = pipe(
|
| 29 |
inputs,
|
|
|
|
| 30 |
generate_kwargs={"task": task},
|
| 31 |
return_timestamps=True,
|
| 32 |
)["text"]
|