bomolopuu commited on
Commit
616e430
·
1 Parent(s): 94769f6
Files changed (1) hide show
  1. asr.py +3 -3
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, AutoProcessor, AutoModelForPreTraining
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/wav2vec2-large-xlsr-53"
23
 
24
  processor = AutoProcessor.from_pretrained(MODEL_ID)
25
- model = AutoModelForPreTraining.from_pretrained(MODEL_ID)
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: