update scatter

app.py

@@ -108,7 +108,7 @@ def main():
         target_label_id = 1
     prompt = st.text_input(TEXT_PROMPT_LABEL[language], PROMPT_EXAMPLE[language])
     alpha = st.slider("Alpha", min_value=-10, max_value=10, step=1, value=0)
-    entropy_threshold = st.slider("Entropy threshold", min_value=0., max_value=5., step=.1, value=2.)
+    entropy_threshold = st.slider("Entropy threshold", min_value=0., max_value=10., step=.1, value=2.)
     plot_idx = np.argmin(np.abs(entropy_threshold - x_s))
     scatter_tip = go.Scatter({
         "x": [x_s[plot_idx]],
@@ -126,7 +126,7 @@ def main():
     })
     figure = go.Figure(data=[scatter, scatter_tip, scatter_tip_lines], layout=layout)
     figure.update_layout(paper_bgcolor="#FFFFFF", plot_bgcolor='#FFFFFF', showlegend=False)
-    st.plotly_chart(figure)
+    st.plotly_chart(figure, use_container_width=True)
     auth_token = os.environ.get('TOKEN') or True
     fp16 = st.checkbox("FP16", value=True)
     with st.spinner('Running inference...'):