added logprobs

app.py

@@ -120,6 +120,7 @@ if __name__ == "__main__":
 
                 avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
 
+        highlighted_out, transition_proba = get_tokens_and_labels
         button.click(get_tokens_and_labels, inputs=prompt, outputs=[highlighted_text, transition_proba])