Your model’s forward method takes one input, while you are passing two into it.
Try to remove batch_size from the model(inputs, batch_size) call and it should work.
PS: you can post code snippets by wrapping them into three backticks ```, which would make debugging a bit easier.