Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -103,5 +103,5 @@ if instruction:
|
|
103 |
|
104 |
st.write(output_text)
|
105 |
|
106 |
-
fig = plot_word_scores(top_token_ids(outputs, threshold
|
107 |
st.plotly_chart(fig, theme=None, use_container_width=False)
|
|
|
103 |
|
104 |
st.write(output_text)
|
105 |
|
106 |
+
fig = plot_word_scores(top_token_ids(outputs, threshold=-10.0), outputs, tokenizer)
|
107 |
st.plotly_chart(fig, theme=None, use_container_width=False)
|