added textbox for logprobs

app.py

@@ -116,7 +116,9 @@ if __name__ == "__main__":
                     show_legend=True,
                 ).style(color_map=label_to_color)
 
-        button.click(get_tokens_and_labels, inputs=prompt, outputs=highlighted_text)
+                avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
+
+        button.click(get_tokens_and_labels, inputs=prompt, outputs=highlighted_text, avg_logprobs)
 
 
 if __name__ == "__main__":