top k removed

app.py

@@ -200,17 +200,17 @@ def main():
         help="The maximum length of the sequence to be generated.",
         min_value=1,
         max_value=MAX_LENGTH,
-        value=int(query_params.get("max_length", [50])[0]),
-        step=1
-    )
-    top_k = st.sidebar.slider(
-        label='Top-k',
-        help="The number of highest probability vocabulary tokens to keep for top-k-filtering",
-        min_value=40,
-        max_value=80,
-        value=int(query_params.get("top_k", [50])[0]),
+        value=int(query_params.get("max_length", [256])[0]),
         step=1
     )
+    # top_k = st.sidebar.slider(
+    #     label='Top-k',
+    #     help="The number of highest probability vocabulary tokens to keep for top-k-filtering",
+    #     min_value=40,
+    #     max_value=80,
+    #     value=int(query_params.get("top_k", [50])[0]),
+    #     step=1
+    # )
     top_p = st.sidebar.slider(
         label='Top-p',
         help="Only the most probable tokens with probabilities that add up to `top_p` or higher are kept for "
@@ -237,7 +237,7 @@ def main():
     )
     generation_kwargs = {
         "max_length": max_length,
-        "top_k": top_k,
+        # "top_k": top_k,
         "top_p": top_p,
         "temperature": temperature,
         "do_sample": do_sample,