musika_api / musika_test.py
nakas's picture
musika clone
050507e
raw
history blame contribute delete
467 Bytes
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
from parse_test import parse_args
from models import Models_functions
from utils import Utils_functions
if __name__ == "__main__":
# parse args
args = parse_args()
# initialize networks
M = Models_functions(args)
models_ls_1, models_ls_2, models_ls_3 = M.get_networks()
# test musika
U = Utils_functions(args)
U.render_gradio(models_ls_1, models_ls_2, models_ls_3, train=False)