nazemi
commited on
Update id_finetuned_model.py
Browse files- id_finetuned_model.py +1 -1
id_finetuned_model.py
CHANGED
|
@@ -2,7 +2,7 @@ from transformers import pipeline
|
|
| 2 |
import sys
|
| 3 |
import time
|
| 4 |
class LI:
|
| 5 |
-
def __init__(self, model_name="
|
| 6 |
|
| 7 |
self.pipe = pipeline("automatic-speech-recognition", model=model_name, return_language=True)
|
| 8 |
def detect_language(self, audio_file_path):
|
|
|
|
| 2 |
import sys
|
| 3 |
import time
|
| 4 |
class LI:
|
| 5 |
+
def __init__(self, model_name="language-identification"):
|
| 6 |
|
| 7 |
self.pipe = pipeline("automatic-speech-recognition", model=model_name, return_language=True)
|
| 8 |
def detect_language(self, audio_file_path):
|