Spaces:
Runtime error
Runtime error
Commit
•
96b8638
1
Parent(s):
27b63b6
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-german
|
| 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-german"
|
| 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-german
|
| 15 |
# make sure no OOM
|
| 16 |
MODEL_NAMES = [
|
| 17 |
-
"bofenghuang/whisper-medium-cv11-german
|
| 18 |
-
"bofenghuang/whisper-large-v2-cv11-german
|
| 19 |
]
|
| 20 |
LANG = "de"
|
| 21 |
CHUNK_LENGTH_S = 30
|
|
|
|
| 11 |
warnings.filterwarnings("ignore")
|
| 12 |
disable_progress_bar()
|
| 13 |
|
| 14 |
+
DEFAULT_MODEL_NAME = "bofenghuang/whisper-large-v2-cv11-german"
|
| 15 |
# make sure no OOM
|
| 16 |
MODEL_NAMES = [
|
| 17 |
+
"bofenghuang/whisper-medium-cv11-german",
|
| 18 |
+
"bofenghuang/whisper-large-v2-cv11-german",
|
| 19 |
]
|
| 20 |
LANG = "de"
|
| 21 |
CHUNK_LENGTH_S = 30
|