Update app.py

app.py

@@ -204,6 +204,14 @@ with st.sidebar:
         step=0.05
     )
 
+    top_k = st.slider(
+        "Top K",
+        min_value=1,
+        max_value=64,
+        value=64,
+        step=1
+    )
+
 # API key input
     api_key = "AIzaSyB93eoXaU48hrJ2hojk-2gQPj28dGVSIbQ"
 
@@ -214,7 +222,7 @@ with st.sidebar:
     generation_config = {
         "temperature": temperature,
         "top_p": top_p,
-        "top_k": 64,
+        "top_k": top_k,
         "max_output_tokens": 8192,
         "response_mime_type": "text/plain",
     }