testing code

app.py

@@ -100,8 +100,6 @@ if __name__ == "__main__":
             """
         )
 
-        highlighted_out, transition_proba = get_tokens_and_labels
-
         with gr.Row():
             with gr.Column():
                 prompt = gr.Textbox(
@@ -120,9 +118,9 @@ if __name__ == "__main__":
                     show_legend=True,
                 ).style(color_map=label_to_color)
 
-                avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
+            gr.Textbox(value = torch.mean(transition_proba))
 
-        button.click(inputs=prompt, outputs=[highlighted_text, transition_proba])
+        button.click(get_tokens_and_labels, inputs=prompt, outputs=[highlighted_text, transition_proba])
 
 
 if __name__ == "__main__":