| # 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) |