b1swas commited on
Commit
243e776
·
verified ·
1 Parent(s): 412d3b8

testing code

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -100,8 +100,6 @@ if __name__ == "__main__":
100
  """
101
  )
102
 
103
- highlighted_out, transition_proba = get_tokens_and_labels
104
-
105
  with gr.Row():
106
  with gr.Column():
107
  prompt = gr.Textbox(
@@ -120,9 +118,9 @@ if __name__ == "__main__":
120
  show_legend=True,
121
  ).style(color_map=label_to_color)
122
 
123
- avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
124
 
125
- button.click(inputs=prompt, outputs=[highlighted_text, transition_proba])
126
 
127
 
128
  if __name__ == "__main__":
 
100
  """
101
  )
102
 
 
 
103
  with gr.Row():
104
  with gr.Column():
105
  prompt = gr.Textbox(
 
118
  show_legend=True,
119
  ).style(color_map=label_to_color)
120
 
121
+ gr.Textbox(value = torch.mean(transition_proba))
122
 
123
+ button.click(get_tokens_and_labels, inputs=prompt, outputs=[highlighted_text, transition_proba])
124
 
125
 
126
  if __name__ == "__main__":