Calling the model requires different args now. Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>