Upload id_finetuned_model.py

id_finetuned_model.py

@@ -0,0 +1,23 @@
+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)
+