Informar de la version de transformers cuando falta AutoModelForTDT
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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(
|