File size: 508 Bytes
6e1a9b7 a6ddb91 6e1a9b7 26540dc 6e1a9b7 fd60f59 6e1a9b7 f17f988 6e1a9b7 fee470d 6e1a9b7 fee470d 645434b 6e1a9b7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# 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) |