# from transformers import AutoConfig, AutoModel
# from custom_model import SpleeterModel

# config = AutoConfig.from_pretrained("niral-env/youtube_spleeter")
# print("----"*30)
# print(config)
# print("----"*30)

# model = SpleeterModel(config)

# print(model)
# result = model.forward("vocals.wav")

# print(result)


from transformers import AutoModel
model = AutoModel.from_pretrained("niral-env/youtube_spleeter", trust_remote_code=True)
print(model)
result = model.forward("vocals.wav")


print(result)