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) | |