It doesn't work for other words

I tried to change the words from "Marvin" into "Follow" or "Yes" or some others. They showed a constant prediction probability at no variation over the inputs. Would you help with the issue please?

Start by running the project's Jupyter notebook and reproducing the prediction behavior for "Marvin," "Follow," and "Yes." Trace how the notebook handles alternate words and inputs; done means those words no longer produce a constant prediction probability, with the expected behavior demonstrated for representative inputs.

Written by the indexing model from the issue text.