Is there a way to make the model output the sampled ligand trajectory coordinates instead of generating an animation? Thanks!