Futuresony commited on
Commit
07c5b0a
·
verified ·
1 Parent(s): 89fc18c

Update asr.py

Browse files
Files changed (1) hide show
  1. asr.py +12 -1
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}"