Spaces:
Sleeping
Sleeping
Commit
·
1c16508
1
Parent(s):
11777ee
up
Browse files- run_demo.py +1 -1
- run_demo_multi_models.py +3 -3
run_demo.py
CHANGED
|
@@ -5,7 +5,7 @@ import pytube as pt
|
|
| 5 |
from transformers import pipeline
|
| 6 |
from huggingface_hub import model_info
|
| 7 |
|
| 8 |
-
MODEL_NAME = "bofenghuang/whisper-medium-cv11-french
|
| 9 |
CHUNK_LENGTH_S = 30
|
| 10 |
|
| 11 |
device = 0 if torch.cuda.is_available() else "cpu"
|
|
|
|
| 5 |
from transformers import pipeline
|
| 6 |
from huggingface_hub import model_info
|
| 7 |
|
| 8 |
+
MODEL_NAME = "bofenghuang/whisper-medium-cv11-french"
|
| 9 |
CHUNK_LENGTH_S = 30
|
| 10 |
|
| 11 |
device = 0 if torch.cuda.is_available() else "cpu"
|
run_demo_multi_models.py
CHANGED
|
@@ -11,11 +11,11 @@ from transformers.utils.logging import disable_progress_bar
|
|
| 11 |
warnings.filterwarnings("ignore")
|
| 12 |
disable_progress_bar()
|
| 13 |
|
| 14 |
-
DEFAULT_MODEL_NAME = "bofenghuang/whisper-large-v2-cv11-french
|
| 15 |
# make sure no OOM
|
| 16 |
MODEL_NAMES = [
|
| 17 |
-
"bofenghuang/whisper-medium-cv11-french
|
| 18 |
-
"bofenghuang/whisper-large-v2-cv11-french
|
| 19 |
]
|
| 20 |
CHUNK_LENGTH_S = 30
|
| 21 |
MAX_NEW_TOKENS = 225
|
|
|
|
| 11 |
warnings.filterwarnings("ignore")
|
| 12 |
disable_progress_bar()
|
| 13 |
|
| 14 |
+
DEFAULT_MODEL_NAME = "bofenghuang/whisper-large-v2-cv11-french"
|
| 15 |
# make sure no OOM
|
| 16 |
MODEL_NAMES = [
|
| 17 |
+
"bofenghuang/whisper-medium-cv11-french",
|
| 18 |
+
"bofenghuang/whisper-large-v2-cv11-french",
|
| 19 |
]
|
| 20 |
CHUNK_LENGTH_S = 30
|
| 21 |
MAX_NEW_TOKENS = 225
|