stable?
Browse files
asr.py
CHANGED
@@ -4,7 +4,7 @@ import os
|
|
4 |
import logging
|
5 |
from pathlib import Path
|
6 |
import torch
|
7 |
-
from transformers import Wav2Vec2ForCTC, AutoProcessor
|
8 |
import numpy as np
|
9 |
|
10 |
# Настройка логирования
|
@@ -19,10 +19,10 @@ with open(f"data/asr/all_langs.tsv") as f:
|
|
19 |
iso, name = line.split(" ", 1)
|
20 |
ASR_LANGUAGES[iso.strip()] = name.strip()
|
21 |
|
22 |
-
MODEL_ID = "facebook/
|
23 |
|
24 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
25 |
-
model =
|
26 |
|
27 |
def safe_process_file(file_obj):
|
28 |
try:
|
|
|
4 |
import logging
|
5 |
from pathlib import Path
|
6 |
import torch
|
7 |
+
from transformers import Wav2Vec2ForCTC, AutoProcessor
|
8 |
import numpy as np
|
9 |
|
10 |
# Настройка логирования
|
|
|
19 |
iso, name = line.split(" ", 1)
|
20 |
ASR_LANGUAGES[iso.strip()] = name.strip()
|
21 |
|
22 |
+
MODEL_ID = "facebook/mms-1b-all"
|
23 |
|
24 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
25 |
+
model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
|
26 |
|
27 |
def safe_process_file(file_obj):
|
28 |
try:
|