This python script provides a user-friendly Gradio interface for inspecting logits produced by the forward pass of language models after applying the softmax function. It allows you to input text and visualize the probability distribution over the vocabulary for each token in the sequence.
