from utils import hf_login, data, compute_wer
from dataset import Dataset
from model import Model

hf_login()






def run_tests (dataset_choice:str, model:str):

    MoDeL = Model()
    MoDeL.select(model)
    MoDeL.load()
    DaTaSeT = Dataset(100)
    DaTaSeT.load(dataset_choice)
    references, predictions = MoDeL.process(DaTaSeT)
    wer = compute_wer(references=references, predictions=predictions)
    return wer


print("WER:", run_tests(dataset_choice="GigaSpeech", model="facebook/s2t-medium-librispeech-asr"))