| import os | |
| from .pipeline import ASR_Diarization | |
| HF_TOKEN = os.environ.get("HF_TOKEN", None) | |
| pipe = ASR_Diarization(HF_TOKEN) | |
| def inference(inputs): | |
| return pipe(inputs) | |
| def inference_with_eval(inputs, output_dir, base_name, ref_rttm=None, ref_json=None): | |
| result = pipe(inputs) | |
| pipe.evaluate(output_dir, base_name, ref_rttm, ref_json) | |
| return result | |