fix

app.py

@@ -141,7 +141,10 @@ def main():
     show_pos_alpha = st.checkbox("Show positive alphas", value=False)
     prompt = st.text_input(TEXT_PROMPT_LABEL[language], PROMPT_EXAMPLE[language])
     st.latex(r"p(x_i|x_{<i}, c) \propto p(x_i|x_{<i})p(c|x_{\leq i})^{\alpha}")
-    alpha = st.slider("α", min_value=-20, max_value=20 if show_pos_alpha else 0, step=1, value=0)
+    if act_type == "softmax":
+        alpha = st.slider("α", min_value=-100, max_value=100 if show_pos_alpha else 0, step=1, value=0)
+    else:
+        alpha = st.slider("α", min_value=-10, max_value=10 if show_pos_alpha else 0, step=1, value=0)
     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({