fix

app.py

@@ -84,7 +84,7 @@ def main():
         target_label_id = 0
     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=0.)
+    entropy_threshold = st.slider("Entropy threshold", min_value=0., max_value=5., step=.1, value=5.)
     auth_token = os.environ.get('TOKEN') or True
     with st.spinner('Running inference...'):
         text = inference(
@@ -92,7 +92,8 @@ def main():
             cls_model_name=cls_model_name,
             prompt=prompt,
             alpha=alpha,
-            entropy_threshold=entropy_threshold
+            target_label_id=target_label_id,
+            entropy_threshold=entropy_threshold,
         )
     st.subheader("Generated text:")
     st.markdown(text)