Update app.py

app.py

@@ -82,14 +82,14 @@ def adjust_distribution(temperature, top_k, top_p, initial_probs):
 
 initial_probs = get_initial_distribution()
 
-def update_plot(temperature, top_k, top_p):
+def update_plot(temperature=1.0, top_k=8, top_p=0.9):
     return adjust_distribution(temperature, top_k, top_p, initial_probs)
 
 interface = gr.Interface(
     fn=update_plot,
     inputs=[
         gr.Slider(0, 5.0, step=0.1, value=1.0, label="Temperature"),
-        gr.Slider(0, 10, step=1, value=5, label="Top-k"),
+        gr.Slider(0, 10, step=1, value=8, label="Top-k"),
         gr.Slider(0.0, 1.0, step=0.01, value=0.9, label="Top-p"),
     ],
     outputs=gr.Plot(label="Token Probability Distribution"),