Deploying the trained NTM model and running the inference