Spaces:
Runtime error
Runtime error
added avg logprobs
Browse files
app.py
CHANGED
|
@@ -118,7 +118,7 @@ if __name__ == "__main__":
|
|
| 118 |
|
| 119 |
avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
|
| 120 |
|
| 121 |
-
button.click(get_tokens_and_labels, inputs=prompt, outputs=highlighted_text, avg_logprobs)
|
| 122 |
|
| 123 |
|
| 124 |
if __name__ == "__main__":
|
|
|
|
| 118 |
|
| 119 |
avg_logprobs = gr.Textbox(value = torch.mean(transition_proba))
|
| 120 |
|
| 121 |
+
button.click(get_tokens_and_labels, inputs=prompt, outputs=[highlighted_text, avg_logprobs])
|
| 122 |
|
| 123 |
|
| 124 |
if __name__ == "__main__":
|