Great work! Can you share a code example of doing inference with your model from the drive on a small dialogue?