Update app.py

app.py

@@ -100,6 +100,8 @@ if __name__ == "__main__":
             """
         )
 
+        highlighted_out, transition_proba = get_tokens_and_labels
+
         with gr.Row():
             with gr.Column():
                 prompt = gr.Textbox(
@@ -120,8 +122,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])
+        button.click(inputs=prompt, outputs=[highlighted_text, transition_proba])
 
 
 if __name__ == "__main__":