Add Language Detector

hubconf.py

@@ -3,6 +3,7 @@ import torch
 from utils import (init_jit_model,
                    get_speech_ts,
                    get_number_ts,
+                   get_language,
                    save_audio,
                    read_audio,
                    state_generator,
@@ -29,7 +30,7 @@ def silero_vad(**kwargs):
 
 
 def silero_number_detector(**kwargs):
-    """Silero Number Detector and Language Classifier
+    """Silero Number Detector
     Returns a model with a set of utils
     Please see https://github.com/snakers4/silero-vad for usage examples
     """
@@ -42,3 +43,16 @@ def silero_number_detector(**kwargs):
              drop_chunks)
 
     return model, utils
+
+
+def silero_lang_detector(**kwargs):
+    """Silero Language Classifier
+    Returns a model with a set of utils
+    Please see https://github.com/snakers4/silero-vad for usage examples
+    """
+    hub_dir = torch.hub.get_dir()
+    model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/number_detector.jit')
+    utils = (get_language,
+             read_audio)
+
+    return model, utils