asr_diarization/inference.py

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