Update app.py

app.py

@@ -85,6 +85,10 @@ initial_probs = get_initial_distribution()
 def update_plot(temperature=1.0, top_k=8, top_p=0.9):
     return adjust_distribution(temperature, top_k, top_p, initial_probs)
 
+# Generate an initial plot with default values
+initial_plot = update_plot()
+
+
 interface = gr.Interface(
     fn=update_plot,
     inputs=[
@@ -92,7 +96,7 @@ interface = gr.Interface(
         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"),
+    outputs=gr.Plot(value=initial_plot, label="Token Probability Distribution"),
     live=True,
     title="Explore generation parameters of LLMs",
     description=description,