Spaces:
Sleeping
Sleeping
Update download_model.py
Browse files- download_model.py +2 -7
download_model.py
CHANGED
@@ -1,16 +1,12 @@
|
|
1 |
# download_model.py
|
2 |
from faster_whisper import WhisperModel
|
3 |
|
4 |
-
#
|
5 |
-
MODEL_NAME = "
|
6 |
|
7 |
print(f"A descarregar e a fazer cache do modelo Whisper '{MODEL_NAME}' no formato CTranslate2...")
|
8 |
|
9 |
try:
|
10 |
-
# Esta chamada é suficiente para descarregar o modelo, convertê-lo
|
11 |
-
# para o formato CTranslate2 e salvá-lo no diretório de cache
|
12 |
-
# definido pela variável de ambiente HF_HUB_CACHE.
|
13 |
-
# O teste de transcrição foi removido para tornar o build mais robusto.
|
14 |
WhisperModel(
|
15 |
MODEL_NAME,
|
16 |
device="cpu",
|
@@ -20,5 +16,4 @@ try:
|
|
20 |
|
21 |
except Exception as e:
|
22 |
print(f"ERRO: Falha ao descarregar ou converter o modelo: {e}")
|
23 |
-
# Faz com que o build falhe se o download do modelo não for bem-sucedido.
|
24 |
exit(1)
|
|
|
1 |
# download_model.py
|
2 |
from faster_whisper import WhisperModel
|
3 |
|
4 |
+
# Usando o modelo 'medium' para máxima precisão.
|
5 |
+
MODEL_NAME = "medium"
|
6 |
|
7 |
print(f"A descarregar e a fazer cache do modelo Whisper '{MODEL_NAME}' no formato CTranslate2...")
|
8 |
|
9 |
try:
|
|
|
|
|
|
|
|
|
10 |
WhisperModel(
|
11 |
MODEL_NAME,
|
12 |
device="cpu",
|
|
|
16 |
|
17 |
except Exception as e:
|
18 |
print(f"ERRO: Falha ao descarregar ou converter o modelo: {e}")
|
|
|
19 |
exit(1)
|