Maximofn commited on
Commit
d996484
·
verified ·
1 Parent(s): f3907f1

Informar de la version de transformers cuando falta AutoModelForTDT

Browse files
Files changed (1) hide show
  1. stt_models.py +10 -1
stt_models.py CHANGED
@@ -95,7 +95,16 @@ def detect_language(audio_file, device):
95
 
96
  def transcribe_parakeet(audio_file, device):
97
  """Transcribe con Parakeet TDT v3, devolviendo timestamps de palabra."""
98
- from transformers import AutoModelForTDT, AutoProcessor
 
 
 
 
 
 
 
 
 
99
 
100
  processor = AutoProcessor.from_pretrained(PARAKEET_MODEL)
101
  model = AutoModelForTDT.from_pretrained(
 
95
 
96
  def transcribe_parakeet(audio_file, device):
97
  """Transcribe con Parakeet TDT v3, devolviendo timestamps de palabra."""
98
+ try:
99
+ from transformers import AutoModelForTDT, AutoProcessor
100
+ except ImportError as exc:
101
+ # AutoModelForTDT llegó con transformers 5.x. Sin él Parakeet no puede
102
+ # cargarse, y conviene que el error diga qué versión hay instalada.
103
+ import transformers
104
+ raise ImportError(
105
+ f"Parakeet necesita AutoModelForTDT (transformers 5.x). "
106
+ f"Instalada: {transformers.__version__}. Detalle: {exc}"
107
+ ) from exc
108
 
109
  processor = AutoProcessor.from_pretrained(PARAKEET_MODEL)
110
  model = AutoModelForTDT.from_pretrained(