sbenfenatti commited on
Commit
10c8fe3
·
verified ·
1 Parent(s): 1f357e0

Update download_model.py

Browse files
Files changed (1) hide show
  1. 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
- # Este script descarrega e converte o modelo durante o 'build' do Docker.
5
- MODEL_NAME = "small"
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)