Spaces:
Sleeping
Sleeping
Update asr.py
Browse files
asr.py
CHANGED
@@ -1,3 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
|
2 |
model.eval()
|
3 |
|
@@ -46,4 +57,4 @@ def transcribe_auto(audio_data=None):
|
|
46 |
final_transcription = processor.decode(ids)
|
47 |
|
48 |
return f"Detected Language: {detect
|
49 |
-
ed_lang.upper()}\n\nTranscription:\n{final_transcription}"
|
|
|
1 |
+
import librosa
|
2 |
+
import torch
|
3 |
+
import numpy as np
|
4 |
+
import langid # Language detection library
|
5 |
+
from transformers import Wav2Vec2ForCTC, AutoProcessor
|
6 |
+
|
7 |
+
ASR_SAMPLING_RATE = 16_000
|
8 |
+
MODEL_ID = "facebook/mms-1b-all"
|
9 |
+
|
10 |
+
# Load MMS Model
|
11 |
+
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
12 |
model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
|
13 |
model.eval()
|
14 |
|
|
|
57 |
final_transcription = processor.decode(ids)
|
58 |
|
59 |
return f"Detected Language: {detect
|
60 |
+
ed_lang.upper()}\n\nTranscription:\n{final_transcription}"
|