utils.py

import argparse
from distutils.util import strtobool

def parse_args():
    
    parser = argparse.ArgumentParser()
    
    # env arguments
    parser.add_argument('--model', type = str, default = 'whisper_medium_baseline_1', help = 'whisper model for inference')
    
    # consolidate args
    args = parser.parse_args()
    args = vars(args)
    
    return args