language-identification / id_finetuned_model.py
nazemi
Update id_finetuned_model.py
47760b6 verified
raw
history blame contribute delete
742 Bytes
from transformers import pipeline
import sys
import time
class LI:
def __init__(self, model_name="language-identification"):
self.pipe = pipeline("automatic-speech-recognition", model=model_name, return_language=True)
def detect_language(self, audio_file_path):
result = self.pipe(audio_file_path)
if 'chunks' in result and result['chunks']:
language = result['chunks'][0]['language']
else:
language = result.get('language', 'Unknown')
return language
if __name__ == "__main__":
li = LI()
audio_file_path = sys.argv[1]
s =time.time()
language = li.detect_language(audio_file_path)
e =time.time()
print(f"Detected language: {language}", e -s)